結果

問題 No.1172 Add Recursive Sequence
ユーザー chineristACchineristAC
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 40 ms
52,224 KB
testcase_01 AC 40 ms
52,352 KB
testcase_02 AC 41 ms
52,224 KB
testcase_03 AC 42 ms
52,608 KB
testcase_04 AC 41 ms
52,480 KB
testcase_05 AC 40 ms
52,608 KB
testcase_06 AC 121 ms
78,096 KB
testcase_07 AC 119 ms
77,492 KB
testcase_08 AC 110 ms
76,800 KB
testcase_09 AC 120 ms
77,344 KB
testcase_10 AC 146 ms
78,720 KB
testcase_11 AC 147 ms
79,104 KB
testcase_12 AC 146 ms
78,336 KB
testcase_13 AC 142 ms
78,208 KB
testcase_14 AC 330 ms
87,488 KB
testcase_15 AC 290 ms
85,796 KB
testcase_16 AC 1,479 ms
87,816 KB
testcase_17 AC 1,391 ms
84,872 KB
権限があれば一括ダウンロードができます

ソースコード

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