結果

問題 No.510 二次漸化式
ユーザー lam6er
提出日時 2025-04-16 00:51:05
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,164 bytes
コンパイル時間 228 ms
コンパイル使用メモリ 82,052 KB
実行使用メモリ 81,700 KB
最終ジャッジ日時 2025-04-16 00:53:31
合計ジャッジ時間 23,362 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 21 TLE * 1 -- * 12
権限があれば一括ダウンロードができます

ソースコード

diff #

MOD = 10**9 + 7

def main():
    import sys
    input = sys.stdin.read().split()
    ptr = 0
    n = int(input[ptr])
    ptr += 1
    q = int(input[ptr])
    ptr += 1

    x = [0] * n
    y = [0] * n

    a = [1] * (n + 1)
    b = [1] * (n + 1)

    for i in range(n):
        b[i+1] = (y[i] * b[i] + 1) % MOD
        a[i+1] = (a[i] + x[i] * (b[i] * b[i] % MOD)) % MOD

    for _ in range(q):
        query = input[ptr]
        ptr += 1
        if query == 'x':
            i = int(input[ptr])
            ptr += 1
            v = int(input[ptr])
            ptr += 1
            x[i] = v
            for j in range(i+1, n+1):
                a[j] = (a[j-1] + x[j-1] * (b[j-1] * b[j-1] % MOD)) % MOD
        elif query == 'y':
            i = int(input[ptr])
            ptr += 1
            v = int(input[ptr])
            ptr += 1
            y[i] = v
            for j in range(i+1, n+1):
                b[j] = (y[j-1] * b[j-1] + 1) % MOD
                a[j] = (a[j-1] + x[j-1] * (b[j-1] * b[j-1] % MOD)) % MOD
        elif query == 'a':
            i = int(input[ptr])
            ptr += 1
            print(a[i] % MOD)

if __name__ == '__main__':
    main()
0