結果

問題 No.876 Range Compress Query
ユーザー lloyzlloyz
提出日時 2022-09-03 18:37:48
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 293 ms / 2,000 ms
コード長 1,521 bytes
コンパイル時間 139 ms
コンパイル使用メモリ 82,472 KB
実行使用メモリ 92,340 KB
最終ジャッジ日時 2024-04-28 16:01:14
合計ジャッジ時間 4,656 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,940 KB
testcase_01 AC 54 ms
66,240 KB
testcase_02 AC 42 ms
58,308 KB
testcase_03 AC 57 ms
67,072 KB
testcase_04 AC 39 ms
54,768 KB
testcase_05 AC 45 ms
61,188 KB
testcase_06 AC 55 ms
66,636 KB
testcase_07 AC 53 ms
66,436 KB
testcase_08 AC 53 ms
66,384 KB
testcase_09 AC 51 ms
65,248 KB
testcase_10 AC 51 ms
64,136 KB
testcase_11 AC 280 ms
88,520 KB
testcase_12 AC 250 ms
86,376 KB
testcase_13 AC 253 ms
88,100 KB
testcase_14 AC 277 ms
87,968 KB
testcase_15 AC 225 ms
89,584 KB
testcase_16 AC 272 ms
92,236 KB
testcase_17 AC 269 ms
92,340 KB
testcase_18 AC 293 ms
91,940 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