結果

問題 No.879 Range Mod 2 Query
ユーザー maspymaspy
提出日時 2020-04-30 19:37:33
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,691 ms / 3,000 ms
コード長 3,700 bytes
コンパイル時間 177 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 145,452 KB
最終ジャッジ日時 2024-05-09 19:15:46
合計ジャッジ時間 16,766 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,992 KB
testcase_01 AC 133 ms
76,976 KB
testcase_02 AC 162 ms
78,400 KB
testcase_03 AC 153 ms
77,612 KB
testcase_04 AC 170 ms
78,556 KB
testcase_05 AC 109 ms
76,748 KB
testcase_06 AC 69 ms
74,120 KB
testcase_07 AC 148 ms
78,316 KB
testcase_08 AC 168 ms
78,268 KB
testcase_09 AC 103 ms
76,916 KB
testcase_10 AC 123 ms
77,392 KB
testcase_11 AC 1,604 ms
141,648 KB
testcase_12 AC 1,168 ms
122,108 KB
testcase_13 AC 1,245 ms
127,728 KB
testcase_14 AC 1,226 ms
124,832 KB
testcase_15 AC 1,257 ms
123,060 KB
testcase_16 AC 1,582 ms
138,684 KB
testcase_17 AC 1,628 ms
139,880 KB
testcase_18 AC 1,691 ms
145,452 KB
testcase_19 AC 842 ms
125,828 KB
testcase_20 AC 839 ms
124,724 KB
testcase_21 AC 866 ms
127,760 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines

class LazySegTree:
    def __init__(self, N, X_f, X_unit, A_f, A_unit, operate):
        self.N = N
        self.X_f = X_f
        self.X_unit = X_unit
        self.A_f = A_f
        self.A_unit = A_unit
        self.operate = operate
        self.X = [self.X_unit] * (N + N)
        self.A = [self.A_unit] * (N + N)
        self.N_elem = [0] * (N+N)
        for i in range(N, N+N):
            self.N_elem[i] = 1
        for i in range(N-1, 0, -1):
            self.N_elem[i] = self.N_elem[i<<1] + self.N_elem[i<<1|1]

    def build(self, seq):
        for i, x in enumerate(seq, N):
            self.X[i] = x
        for i in range(N - 1, 0, -1):
            self.X[i] = self.X_f(self.X[i << 1], self.X[i << 1 | 1])

    def _eval_at(self, i):
        return self.operate(self.X[i], self.A[i], self.N_elem[i])

    def _propagate_at(self, i):
        self.X[i] = self._eval_at(i)
        self.A[i << 1] = self.A_f(self.A[i << 1], self.A[i])
        self.A[i << 1 | 1] = self.A_f(self.A[i << 1 | 1], self.A[i])
        self.A[i] = self.A_unit

    def _propagate_above(self, i):
        H = i.bit_length() - 1
        for h in range(H, 0, -1):
            self._propagate_at(i >> h)

    def _recalc_above(self, i):
        while i > 1:
            i >>= 1
            self.X[i] = self.X_f(self._eval_at(i << 1),
                                 self._eval_at(i << 1 | 1))

    def set_val(self, i, x):
        i += self.N
        self._propagate_above(i)
        self.X[i] = x
        self.A[i] = self.A_unit
        self._recalc_above(i)

    def operate_range(self, L, R, x):
        L += self.N
        R += self.N
        L0 = L // (L & -L)
        R0 = R // (R & -R) - 1
        self._propagate_above(L0)
        self._propagate_above(R0)
        while L < R:
            if L & 1:
                self.A[L] = self.A_f(self.A[L], x)
                L += 1
            if R & 1:
                R -= 1
                self.A[R] = self.A_f(self.A[R], x)
            L >>= 1
            R >>= 1
        self._recalc_above(L0)
        self._recalc_above(R0)

    def fold(self, L, R):
        L += self.N
        R += self.N
        self._propagate_above(L // (L & -L))
        self._propagate_above(R // (R & -R) - 1)
        v = self.X_unit
        while L < R:
            if L & 1:
                v = self.X_f(v, self._eval_at(L))
                L += 1
            if R & 1:
                R -= 1
                v = self.X_f(self._eval_at(R), v)
            L >>= 1
            R >>= 1
        return v

X_unit = (0, 0)  # odd, sum
A_unit = (2, 0)


def X_f(x, y):
    return (x[0] + y[0], x[1] + y[1])


def A_f(x, y):
    if y[0] <= 1:
        return ((x[0] + x[1] + y[0]) & 1, y[1])
    return (x[0], x[1] + y[1])


def operate(x, a, n):
    od, su = x
    t, add = a
    if t == 2:
        if add & 1:
            od = n - od
        return (od, su + add * n)
    if t == 1:
        od = n - od
    su = od + add * n
    if add & 1:
        od = n - od
    return (od, su)

N, Q = map(int, readline().split())
seg = LazySegTree(N, X_f, X_unit, A_f, A_unit, operate)
A = map(int, readline().split())
A = ((1, x) if x & 1 else (0, x) for x in A)
seg.build(A)

for query in readlines():
    if query.startswith(b'1'):
        L, R = map(int, query[2:].split())
        seg.operate_range(L - 1, R, (0, 0))
    elif query.startswith(b'2'):
        L, R, x = map(int, query[2:].split())
        seg.operate_range(L - 1, R, (2, x))
    elif query.startswith(b'3'):
        L, R = map(int, query[2:].split())
        ev = 0
        od = 0
        print(seg.fold(L - 1, R)[1])
0