結果

問題 No.1172 Add Recursive Sequence
ユーザー chineristACchineristAC
提出日時 2020-08-15 04:47:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,400 ms / 4,000 ms
コード長 1,325 bytes
コンパイル時間 247 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 93,440 KB
最終ジャッジ日時 2024-10-10 17:10:44
合計ジャッジ時間 7,402 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
52,736 KB
testcase_01 AC 37 ms
52,608 KB
testcase_02 AC 37 ms
52,608 KB
testcase_03 AC 36 ms
52,224 KB
testcase_04 AC 38 ms
52,224 KB
testcase_05 AC 37 ms
52,224 KB
testcase_06 AC 110 ms
77,620 KB
testcase_07 AC 105 ms
77,408 KB
testcase_08 AC 101 ms
76,944 KB
testcase_09 AC 107 ms
77,556 KB
testcase_10 AC 130 ms
79,140 KB
testcase_11 AC 130 ms
78,720 KB
testcase_12 AC 134 ms
78,464 KB
testcase_13 AC 129 ms
78,544 KB
testcase_14 AC 292 ms
87,576 KB
testcase_15 AC 257 ms
85,268 KB
testcase_16 AC 2,137 ms
93,440 KB
testcase_17 AC 2,400 ms
88,884 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys

input=sys.stdin.buffer.readline

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

check=[first[i] for i in range(K)]
for i in range(1,K):
    check[i]+=sum(c[j]*check[i-1-j] for j in range(i))
    check[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