Submission #61563119

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
import sys
def main():
    N = int(sys.stdin.readline())
    A = list(map(int, sys.stdin.readline().split()))
    diff = [0] * (N + 2)
    B = [0] * N
    curSum = 0
    for j in range(1, N + 1):
        curSum += diff[j]
        total = A[j - 1] + curSum
        tmp = min(total, N - j)
        B[j - 1] = total - tmp
        if tmp > 0:
            start = j + 1
            end = j + tmp + 1
            if start <= N:
                diff[start] += 1
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

import sys


def main():
    N = int(sys.stdin.readline())
    A = list(map(int, sys.stdin.readline().split()))

    diff = [0] * (N + 2)
    B = [0] * N
    curSum = 0

    for j in range(1, N + 1):
        curSum += diff[j]
        total = A[j - 1] + curSum
        tmp = min(total, N - j)
        B[j - 1] = total - tmp
        if tmp > 0:
            start = j + 1
            end = j + tmp + 1
            if start <= N:
                diff[start] += 1
            if end <= N:
                diff[end] -= 1
    print(" ".join(map(str, B)))


if __name__ == "__main__":
    main()

Submission Info

Judge Result