結果

問題 No.1172 Add Recursive Sequence
ユーザー chineristAC
提出日時 2020-08-15 04:56:31
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,479 ms / 4,000 ms
コード長 1,299 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 87,816 KB
最終ジャッジ日時 2024-10-10 17:10:50
合計ジャッジ時間 6,148 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

input=sys.stdin.buffer.readline

def sum(iter):
    res=0
    mod=10**9+7
    for a in iter:
        res+=a
        res%=mod
    return res

K,N,M=map(int,input().split())
a=list(map(int,input().split()))
c=list(map(int,input().split()))
mod=10**9+7

seq=a+[0]*(N-K)
for i in range(K,N):
    seq[i]=sum(c[j]*seq[i-1-j] for j in range(K))
    seq[i]%=mod

end_data=[0 for i in range(K)]

first=[a[i] for i in range(K)]
for i in range(1,K):
    minus=sum(a[j]*c[i-1-j] for j in range(i))
    first[i]-=minus
    first[i]%=mod

x=[0]*(N+K+1)
too_short=[0]*N
end_query=[[] for i in range(N)]

for _ in range(M):
    l,r=map(int,input().split())
    if r-l>=K:
        for i in range(l,l+K):
            x[i]+=first[i-l]
            x[i]%=mod
        if r-l!=N:
            end_query[r-l].append(r)
    else:
        for i in range(l,r):
            too_short[i]+=a[i-l]
            too_short[i]%=mod

for i in range(N-1,-1,-1):
    for j in range(K-1,-1,-1):
        end_data[j]=(end_data[j-1]-c[j-1]*seq[i])%mod
    end_data[0]=seq[i]
    for r in end_query[i]:
        for j in range(r,r+K):
            x[j]-=end_data[j-r]
            x[j]%=mod

for i in range(N):
    x[i]+=sum(c[j]*x[i-1-j] for j in range(min(K,i)))
    x[i]%=mod

for i in range(N):
    print((x[i]+too_short[i])%mod)
0