結果

問題 No.1102 Remnants
コンテスト
ユーザー FromBooska
提出日時 2023-03-22 17:48:56
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 915 bytes
コンパイル時間 556 ms
コンパイル使用メモリ 82,264 KB
実行使用メモリ 253,496 KB
最終ジャッジ日時 2024-09-18 15:00:07
合計ジャッジ時間 4,036 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 TLE * 1
other -- * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

# 寄与数をどうやって求めるのかわからず
# 公式解説より
# K回の操作、各回の(l, r)を(li, ri)とすれば
# 1 <= l1 <= l2 <= ---- lK <= rK <= ---- <= r1 <= N
# あるAi項がK回後に残っているには lK <= i <= rK
# 左側は nCk(i+K-1, K)
# 右側は nCk(N-i+K, K)
# その積が寄与分となる

mod = 10**9+7
N, K = map(int, input().split())
A = list(map(int, input().split()))
 
# nCrメモ化パッケージ
factorial = [1] #0分
inverse = [1] #0分
for i in range(1, N+K+1):
    factorial.append(factorial[-1]*i%mod)
    inverse.append(pow(factorial[-1], mod-2, mod))
    
def nCr_fast(N, R, MOD):
    if N < R or R < 0:
        return 0
    elif R == 0 or R == N:
        return 1
    return factorial[N]*inverse[R]*inverse[N-R]%MOD

ans = 0
for i in range(N):
    count = nCr_fast(i+K, K, mod) * nCr_fast(N-1-i+K, K, mod)
    ans += count*A[i]
    ans %= mod
print(ans)
0