結果

問題 No.1102 Remnants
ユーザー hedwig100hedwig100
提出日時 2020-07-03 22:25:41
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 430 ms / 2,000 ms
コード長 788 bytes
コンパイル時間 1,145 ms
コンパイル使用メモリ 81,468 KB
実行使用メモリ 103,276 KB
最終ジャッジ日時 2023-10-17 05:16:53
合計ジャッジ時間 6,137 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 37 ms
53,352 KB
testcase_01 AC 38 ms
53,352 KB
testcase_02 AC 37 ms
53,352 KB
testcase_03 AC 37 ms
53,352 KB
testcase_04 AC 37 ms
53,352 KB
testcase_05 AC 311 ms
97,568 KB
testcase_06 AC 89 ms
68,288 KB
testcase_07 AC 430 ms
101,252 KB
testcase_08 AC 54 ms
63,960 KB
testcase_09 AC 62 ms
63,960 KB
testcase_10 AC 51 ms
63,964 KB
testcase_11 AC 63 ms
68,076 KB
testcase_12 AC 69 ms
70,124 KB
testcase_13 AC 61 ms
63,960 KB
testcase_14 AC 192 ms
82,136 KB
testcase_15 AC 157 ms
76,988 KB
testcase_16 AC 359 ms
101,860 KB
testcase_17 AC 379 ms
101,396 KB
testcase_18 AC 380 ms
103,276 KB
testcase_19 AC 167 ms
83,472 KB
testcase_20 AC 83 ms
73,976 KB
testcase_21 AC 247 ms
91,948 KB
testcase_22 AC 330 ms
100,144 KB
testcase_23 AC 266 ms
92,316 KB
testcase_24 AC 204 ms
86,536 KB
testcase_25 AC 382 ms
103,272 KB
testcase_26 AC 70 ms
70,116 KB
testcase_27 AC 141 ms
81,164 KB
権限があれば一括ダウンロードができます

ソースコード

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