結果

問題 No.1102 Remnants
ユーザー hedwig100
提出日時 2020-07-03 22:25:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 449 ms / 2,000 ms
コード長 788 bytes
コンパイル時間 286 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 103,564 KB
最終ジャッジ日時 2024-09-17 03:58:21
合計ジャッジ時間 6,284 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
sys.setrecursionlimit(10000000)
MOD = 10 ** 9 + 7
INF = 10 ** 15

def nck_nbig(n,r,mod = MOD):
    r = min(n,r)
    if r == 0:
        return 1
    num,inv_num = 1,1
    for i in range(1,r + 1):
        num = num * (n - i + 1) %mod
        inv_num = inv_num * i % mod
    return num * pow(inv_num,mod - 2,mod)%mod

def main():
	N,K = map(int,input().split())
	A = list(map(int,input().split()))
    

	ans = 0
	before_nck1 = 1
	before_nck2 = nck_nbig(N + K - 1,N - 1)
	for i in range(N):
		if i > 0:
			before_nck1 *= (K + i)
			before_nck1 *= pow(i,MOD - 2,MOD)
			before_nck1 %= MOD
			before_nck2 *= pow(K + N - i,MOD-2,MOD)
			before_nck2 *= (N - i)
			before_nck2 %= MOD
		ans += before_nck1 * before_nck2 * A[i]
		ans %= MOD
	print(ans)
if __name__ == '__main__':
	main()
0