結果

問題 No.230 Splarraay ｽﾌﾟﾗﾚｪｰｲ
コンテスト
ユーザー maspy
提出日時 2020-04-25 06:02:12
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,431 ms / 5,000 ms
コード長 3,187 bytes
コンパイル時間 990 ms
コンパイル使用メモリ 82,436 KB
実行使用メモリ 208,848 KB
最終ジャッジ日時 2024-11-06 05:26:42
合計ジャッジ時間 16,488 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

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 = None

    @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 x is None:
            return y
        if y is None:
            return x
        return y

    @classmethod
    def operate(cls, x, y):
        if y is None:
            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.H = (self.N - 1).bit_length()
        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):
        for h in range(self.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)
        self._propagate_above(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 range_operate(self, L, R, x):
        L += self.N
        R += self.N
        L0 = L
        R0 = 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.range_operate(L, R, x)

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