結果

問題 No.876 Range Compress Query
ユーザー lloyzlloyz
提出日時 2022-09-03 18:37:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 331 ms / 2,000 ms
コード長 1,521 bytes
コンパイル時間 346 ms
コンパイル使用メモリ 82,020 KB
実行使用メモリ 92,236 KB
最終ジャッジ日時 2024-11-17 09:35:02
合計ジャッジ時間 5,351 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
53,616 KB
testcase_01 AC 62 ms
65,144 KB
testcase_02 AC 43 ms
58,064 KB
testcase_03 AC 64 ms
67,464 KB
testcase_04 AC 45 ms
55,420 KB
testcase_05 AC 47 ms
60,912 KB
testcase_06 AC 65 ms
66,932 KB
testcase_07 AC 60 ms
66,424 KB
testcase_08 AC 60 ms
65,144 KB
testcase_09 AC 58 ms
65,004 KB
testcase_10 AC 58 ms
65,572 KB
testcase_11 AC 324 ms
87,968 KB
testcase_12 AC 286 ms
86,204 KB
testcase_13 AC 290 ms
87,628 KB
testcase_14 AC 310 ms
87,944 KB
testcase_15 AC 263 ms
89,916 KB
testcase_16 AC 316 ms
91,532 KB
testcase_17 AC 315 ms
92,236 KB
testcase_18 AC 331 ms
92,168 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Fenwick_Tree:
    def __init__(self, n):
        self.n = n
        self.data = [0] * n

    def add(self, p, x):
        p += 1
        while p <= self.n:
            self.data[p - 1] += x
            p += p & -p

    def sum(self, l, r):
        '''範囲[l, r)(lからr-1まで)の総和を求める'''
        return self._sum(r) - self._sum(l)

    def _sum(self, r):
        '''範囲[0, r)(0からr-1まで)の総和を求める'''
        s = 0
        while r > 0:
            s += self.data[r - 1]
            r -= r & -r
        return s

n, q = map(int, input().split())
A = list(map(int, input().split()))
Dif = [A[i + 1] - A[i] for i in range(n - 1)]
FenTree = Fenwick_Tree(n - 1)
for i in range(n - 1):
    if Dif[i] != 0:
        FenTree.add(i, 1)
for _ in range(q):
    L = list(map(int, input().split()))
    if L[0] == 1:
        l, r, x = L[1:]
        l -= 2
        if l >= 0:
            prev = Dif[l]
            curr = Dif[l] + x
            if prev == 0:
                FenTree.add(l, 1)
            elif curr == 0:
                FenTree.add(l, -1)
            Dif[l] = curr
        r -= 1
        if r < n - 1:
            prev = Dif[r]
            curr = Dif[r] - x
            if prev == 0:
                FenTree.add(r, 1)
            elif curr == 0:
                FenTree.add(r, -1)
            Dif[r] = curr
    elif L[0] == 2:
        l, r = L[1:]
        if l == r:
            print(1)
            continue
        l -= 1
        r -= 1
        print(1 + FenTree.sum(l, r))
0