結果

問題 No.230 Splarraay スプラレェーイ
ユーザー 👑 maspymaspy
提出日時 2020-04-25 12:00:24
言語 PyPy3
(7.3.1)
結果
AC  
実行時間 2,270 ms / 5,000 ms
コード長 3,164 Byte
コンパイル時間 179 ms
使用メモリ 196,476 KB
最終ジャッジ日時 2020-08-26 08:13:35
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
testcase_00 AC 40 ms
63,884 KB
testcase_01 AC 40 ms
63,716 KB
testcase_02 AC 42 ms
64,392 KB
testcase_03 AC 41 ms
63,044 KB
testcase_04 AC 39 ms
63,760 KB
testcase_05 AC 179 ms
81,660 KB
testcase_06 AC 382 ms
87,312 KB
testcase_07 AC 150 ms
80,936 KB
testcase_08 AC 241 ms
82,564 KB
testcase_09 AC 1,198 ms
132,236 KB
testcase_10 AC 1,007 ms
114,308 KB
testcase_11 AC 855 ms
112,516 KB
testcase_12 AC 1,171 ms
133,080 KB
testcase_13 AC 450 ms
91,240 KB
testcase_14 AC 477 ms
125,904 KB
testcase_15 AC 1,067 ms
125,856 KB
testcase_16 AC 1,507 ms
154,036 KB
testcase_17 AC 2,270 ms
196,476 KB
testcase_18 AC 638 ms
126,740 KB
testcase_19 AC 1,603 ms
159,204 KB
権限があれば一括ダウンロードができます

ソースコード

diff #
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines


class LazySegTree:
    X_unit = (0, 0, 0)  # 無色、色1、色2
    A_unit = 0

    @classmethod
    def X_f(cls, x, y):
        a, b, c = x
        d, e, f = y
        return (a + d, b + e, c + f)

    @classmethod
    def A_f(cls, x, y):
        if not y:
            return x
        return y

    @classmethod
    def operate(cls, x, y):
        if not y:
            return x
        s = sum(x)
        if y == 1:
            return (0, s, 0)
        else:
            return (0, 0, s)

    def __init__(self, N):
        self.N = N
        self.X = [self.X_unit] * (N + N)
        self.A = [self.A_unit] * (N + N)

    def build(self, seq):
        for i, x in enumerate(seq, self.N):
            self.X[i] = x
        for i in range(self.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])

    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 fold(self, L, R):
        L += self.N
        R += self.N
        self._propagate_above(L // (L & -L))
        self._propagate_above(R // (R & -R) - 1)
        vL = self.X_unit
        vR = self.X_unit
        while L < R:
            if L & 1:
                vL = self.X_f(vL, self._eval_at(L))
                L += 1
            if R & 1:
                R -= 1
                vR = self.X_f(self._eval_at(R), vR)
            L >>= 1
            R >>= 1
        return self.X_f(vL, vR)

    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)


N = int(readline())
Q = int(readline())

score_A = 0
score_B = 0
seg = LazySegTree(N)
seg.build([(1, 0, 0)] * N)
m = map(int, read().split())
for x, L, R in zip(m, m, m):
    R += 1
    if x == 0:
        _, a, b = seg.fold(L, R)
        if a > b:
            score_A += a
        elif b > a:
            score_B += b
    else:
        seg.operate_range(L, R, x)

_, a, b = seg.fold(0, N)
print(score_A + a, score_B + b)
0