結果

問題 No.1102 Remnants
ユーザー 👑 Kazun
提出日時 2021-06-16 20:34:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 152 ms / 2,000 ms
コード長 444 bytes
コンパイル時間 414 ms
コンパイル使用メモリ 82,092 KB
実行使用メモリ 114,596 KB
最終ジャッジ日時 2024-12-31 07:25:57
合計ジャッジ時間 3,996 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

N,K=map(int,input().split())
A=["*"]+list(map(int,input().split()))

Mod=10**9+7
Inv=[0]*(N+1); Inv[1]=1
for i in range(2,N+1):
    q,r=divmod(Mod,i)
    Inv[i]=-q*Inv[r]%Mod

L=[0]*(N+1); L[1]=1
for i in range(2,N+1):
    b=(i+K-1)*Inv[i-1]%Mod
    L[i]=b*L[i-1]%Mod

R=[0]*(N+1); R[N]=1
for i in range(N-1,0,-1):
    b=(N-i+K)*Inv[N-i]
    R[i]=b*R[i+1]%Mod

X=0
for a,l,r in zip(A[1:],L[1:],R[1:]):
    X+=a*((l*r)%Mod)
    X%=Mod

print(X)
0