結果

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

ソースコード

diff # 

import sys
import io, os
input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline

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
 
    for j in range(K):
        if r+j<N:
            S[r+j]=(S[r+j]+calc_r(r-l-1,j))%mod

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