結果

問題 No.1172 Add Recursive Sequence
ユーザー titia
提出日時 2024-11-05 04:57:37
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,559 ms / 4,000 ms
コード長 1,448 bytes
コンパイル時間 522 ms
コンパイル使用メモリ 82,396 KB
実行使用メモリ 250,320 KB
最終ジャッジ日時 2024-11-05 04:57:46
合計ジャッジ時間 7,821 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.readline

from functools import lru_cache

K,N,M=map(int,input().split())
A=list(map(int,input().split()))
C=list(map(int,input().split()))

mod=10**9+7

for i in range(N):
    sc=0
    for j in range(K):
        sc+=C[j]*A[-1-j]
        sc%=mod
    A.append(sc)

AC=[[] for i in range(N)]

for i in range(N):
    X=[]
    sc=0
    for j in range(K):
        if i-j>=0:
            X.append(A[i-j]*C[j]%mod)
        else:
            X.append(0)

    X.reverse()

    for j in range(1,len(X)):
        X[j]=(X[j]+X[j-1])%mod

    for j in range(len(X)):
        X[j]=-X[j]%mod

    AC[i]=X
        

CL=[-1]*K
def calc_l(x):
    if CL[x]!=-1:
        return CL[x]
    ANS=A[x]
    for i in range(1,K):
        if x-i>=0:
            ANS=(ANS-A[x-i]*C[i-1])%mod

    CL[x]=ANS
    return ANS

for i in range(K):
    CL[i]=calc_l(i)

def calc_r(stind,x):
    return AC[stind+x][-x-1]


S=[0]*N

for i in range(M):
    l,r=map(int,input().split())
    for j in range(K):
        if l+j<r and l+j<N:
            S[l+j]=(S[l+j]+calc_l(j))%mod

        #print("!",S)
        
    for j in range(K):
        if r+j<N:
            S[r+j]=(S[r+j]+calc_r(r-l-1,j))%mod

        #print("!",S)

    #print(S)

    #for j in range(K):
    #    S[r+j]-=A[j]

ANS=[0]*N

for i in range(N):
    ANS[i]=S[i]%mod
    for j in range(K):
        if i-j-1>=0:
            ANS[i]=(ANS[i]+ANS[i-j-1]*C[j])%mod

print("\n".join(map(str,ANS)))
0