結果

問題 No.1099 Range Square Sum
コンテスト
ユーザー tktk_snsn
提出日時 2020-12-15 20:50:33
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,927 ms / 2,000 ms
コード長 3,933 bytes
コンパイル時間 151 ms
コンパイル使用メモリ 81,920 KB
実行使用メモリ 230,368 KB
最終ジャッジ日時 2024-09-20 02:10:16
合計ジャッジ時間 16,847 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

class LazySegmetTree:
    def __init__(self, size, op_data, u_data, op_lazy, u_lazy, op_merge):
        """
        size: 元の配列の長さ
        op_data(Ldata, Rdata), u_data: クエリ関数、単位元
        op_lazy(Llazy, Rlazy), u_lazy: 遅延作用の関数、単位元
        op_merge(lazy, data): dataとlazyとの演算
        """
        self.N = size
        self.op_data = op_data
        self.op_lazy = op_lazy
        self.op_merge = op_merge
        self.u_data = u_data
        self.u_lazy = u_lazy
        self.data = [u_data] * (2 * self.N)
        self.lazy = [u_lazy] * (2 * self.N)

    def initialize(self, arr):
        op = self.op_data
        for i, a in enumerate(arr, self.N):
            self.data[i] = a
        for i in reversed(range(1, self.N)):
            self.data[i] = op(self.data[i << 1], self.data[i << 1 | 1])

    def show(self):
        print("show array")
        for i in range(self.N):
            print(self.fold(i, i+1), end=" ")
        print("\n")

    def _propagate_top_down(self, L):
        op = self.op_lazy
        merge = self.op_merge
        unit = self.u_lazy
        for i in reversed(range(1, L.bit_length())):
            i = L >> i
            v = self.lazy[i]
            if v == unit:
                continue
            self.data[i] = merge(v, self.data[i])
            self.lazy[i << 1] = op(v, self.lazy[i << 1])
            self.lazy[i << 1 | 1] = op(v, self.lazy[i << 1 | 1])
            self.lazy[i] = unit

    def _propagate_bottom_up(self, i):
        merge = self.op_merge
        op = self.op_data
        while i > 1:
            i >>= 1
            self.data[i] = op(merge(self.lazy[i << 1], self.data[i << 1]),
                              merge(self.lazy[i << 1 | 1], self.data[i << 1 | 1]))

    def fold(self, L, R):
        resL = self.u_data
        resR = self.u_data
        op = self.op_data
        merge = self.op_merge
        L += self.N
        R += self.N
        L0 = L // (L & -L)
        R0 = R // (R & -R) - 1
        self._propagate_top_down(L0)
        self._propagate_top_down(R0)
        while L < R:
            if L & 1:
                resL = op(resL, merge(self.lazy[L], self.data[L]))
                L += 1
            if R & 1:
                R -= 1
                resR = op(merge(self.lazy[R], self.data[R]), resR)
            L >>= 1
            R >>= 1
        return op(resL, resR)

    def update(self, i, val):  # 未調整
        i += self.N
        i0 = i // (i & -i)
        self._propagate_top_down(i0)
        self.data[i] = self.op_merge(self.lazy[i], self.data[i])
        self.data[i] = self.op_data(val, self.data[i])
        self.lazy[i] = self.u_lazy
        self._propagate_bottom_up(i0)

    def range_update(self, L, R, val):
        op = self.op_lazy
        L += self.N
        R += self.N
        L0 = L // (L & -L)
        R0 = R // (R & -R) - 1
        self._propagate_top_down(L0)
        self._propagate_top_down(R0)
        while L < R:
            if L & 1:
                self.lazy[L] = op(val, self.lazy[L])
                L += 1
            if R & 1:
                R -= 1
                self.lazy[R] = op(val, self.lazy[R])
            L >>= 1
            R >>= 1
        self._propagate_bottom_up(L0)
        self._propagate_bottom_up(R0)


u_data = (0, 0, 0)  # size, sum, sqsum
u_lazy = 0


def op_data(l, r):
    return l[0]+r[0], l[1]+r[1], l[2]+r[2]


def op_lazy(a, b):
    return a + b


def op_merge(lz, d):
    return d[0], d[1] + d[0] * lz, d[2] + (2 * d[1] + d[0] * lz) * lz


N = int(input())
A = tuple(map(int, input().split()))

seg = LazySegmetTree(N, op_data, u_data, op_lazy, u_lazy, op_merge)
seg.initialize([(1, a, a * a) for a in A])

Q = int(input())
for _ in range(Q):
    k, *arg = map(int, input().split())
    if k == 1:
        l, r, x = arg
        seg.range_update(l - 1, r, x)
    else:
        l, r = arg
        print(seg.fold(l - 1, r)[2])
0