結果

問題 No.510 二次漸化式
ユーザー maspymaspy
提出日時 2020-04-03 01:16:06
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 761 ms / 3,000 ms
コード長 2,150 bytes
コンパイル時間 1,251 ms
コンパイル使用メモリ 87,044 KB
実行使用メモリ 154,856 KB
最終ジャッジ日時 2023-09-11 01:09:21
合計ジャッジ時間 20,449 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 72 ms
71,588 KB
testcase_01 AC 73 ms
71,372 KB
testcase_02 AC 416 ms
90,692 KB
testcase_03 AC 417 ms
90,508 KB
testcase_04 AC 424 ms
90,688 KB
testcase_05 AC 428 ms
90,816 KB
testcase_06 AC 582 ms
107,608 KB
testcase_07 AC 573 ms
105,832 KB
testcase_08 AC 572 ms
106,192 KB
testcase_09 AC 599 ms
106,224 KB
testcase_10 AC 287 ms
80,612 KB
testcase_11 AC 294 ms
80,768 KB
testcase_12 AC 282 ms
80,780 KB
testcase_13 AC 281 ms
80,152 KB
testcase_14 AC 287 ms
80,436 KB
testcase_15 AC 283 ms
80,388 KB
testcase_16 AC 435 ms
101,988 KB
testcase_17 AC 446 ms
103,152 KB
testcase_18 AC 450 ms
102,192 KB
testcase_19 AC 441 ms
102,080 KB
testcase_20 AC 440 ms
102,916 KB
testcase_21 AC 431 ms
102,148 KB
testcase_22 AC 432 ms
101,964 KB
testcase_23 AC 686 ms
123,500 KB
testcase_24 AC 694 ms
123,480 KB
testcase_25 AC 695 ms
123,948 KB
testcase_26 AC 692 ms
124,140 KB
testcase_27 AC 689 ms
125,064 KB
testcase_28 AC 681 ms
124,120 KB
testcase_29 AC 705 ms
123,660 KB
testcase_30 AC 689 ms
123,900 KB
testcase_31 AC 721 ms
153,824 KB
testcase_32 AC 750 ms
154,476 KB
testcase_33 AC 761 ms
154,856 KB
testcase_34 AC 471 ms
85,400 KB
testcase_35 AC 424 ms
84,604 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/ python3.8
import sys
read = sys.stdin.buffer.read
readline = sys.stdin.buffer.readline
readlines = sys.stdin.buffer.readlines
MOD = 10 ** 9 + 7


def mat(x, y):
    return (1, x, 0, 0, 0, y * y, 2 * y, 1, 0, 0, y, 1, 0, 0, 0, 1)


class SegTree:
    """ segment tree with point modification and range product access. """
    data_unit = mat(0, 0)
    @classmethod
    def data_f(cls, A, B):
        C = [0] * 16
        for i in range(4):
            for j in range(i, 4):
                for k in range(j, 4):
                    C[4 * i + k] += B[4 * i + j] * A[4 * j + k]
        return tuple(x % MOD for x in C)

    def __init__(self, N):
        self.N = N
        self.data = [self.data_unit] * (N + N)

    def build(self, raw_data):
        data = self.data
        f = self.data_f
        N = self.N
        data[N:] = raw_data[:]
        for i in range(N - 1, 0, -1):
            data[i] = f(data[i << 1], data[i << 1 | 1])

    def set_val(self, i, x):
        data = self.data
        f = self.data_f
        i += self.N
        data[i] = x
        while i > 1:
            i >>= 1
            data[i] = f(data[i << 1], data[i << 1 | 1])

    def fold(self, L, R):
        """ compute for [L, R) """
        vL = vR = self.data_unit
        data = self.data
        f = self.data_f
        L += self.N
        R += self.N
        while L < R:
            if L & 1:
                vL = f(vL, data[L])
                L += 1
            if R & 1:
                R -= 1
                vR = f(data[R], vR)
            L >>= 1
            R >>= 1
        return f(vL, vR)


N = int(readline())
X = [0] * N
Y = [0] * N
Q = int(readline())

seg = SegTree(N)


for query in readlines()[:Q]:
    if query.startswith(b'a'):
        i = int(query[2:])
        if i == 0:
            print(1)
        else:
            A = seg.fold(0, i)
            print(sum(A[:4]) % MOD)
    elif query.startswith(b'x'):
        i, v = map(int, query[2:].split())
        X[i] = v
        seg.set_val(i, mat(v, Y[i]))
    elif query.startswith(b'y'):
        i, v = map(int, query[2:].split())
        Y[i] = v
        seg.set_val(i, mat(X[i], v))
0