結果

問題 No.1172 Add Recursive Sequence
ユーザー lam6er
提出日時 2025-03-20 18:49:05
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,503 bytes
コンパイル時間 158 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 80,420 KB
最終ジャッジ日時 2025-03-20 18:50:41
合計ジャッジ時間 10,877 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 12 TLE * 1 -- * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

MOD = 10**9 + 7

def main():
    import sys
    input = sys.stdin.read
    data = input().split()
    ptr = 0
    K = int(data[ptr])
    ptr += 1
    N = int(data[ptr])
    ptr += 1
    M = int(data[ptr])
    ptr += 1

    a = list(map(int, data[ptr:ptr+K]))
    ptr += K
    c = list(map(int, data[ptr:ptr+K]))
    ptr += K

    # Precompute a up to maximum needed index
    max_needed = 0
    queries = []
    for _ in range(M):
        l = int(data[ptr])
        ptr += 1
        r = int(data[ptr])
        ptr += 1
        L = r - l
        queries.append((l, r))
        if L - 1 > max_needed:
            max_needed = L - 1

    # Extend the a array up to max_needed
    for _ in range(len(a), max_needed + 1):
        next_val = 0
        for k in range(1, K+1):
            if len(a) - k >= 0:
                next_val = (next_val + c[k-1] * a[len(a) - k]) % MOD
        a.append(next_val)

    # Initialize x array
    x = [0] * (N + 1)  # Using 1-based indexing for convenience

    # Process each query
    for l, r in queries:
        L = r - l
        # For each position in the query's range
        start = l
        end = min(r, N)
        length = end - start
        for d in range(length):
            if d > max_needed:
                break  # Since we precomputed up to max_needed, which is the maximum of all L-1
            x[start + d] = (x[start + d] + a[d]) % MOD

    # Output the result
    for i in range(N):
        print(x[i] % MOD)

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