結果
問題 |
No.1489 Repeat Cumulative Sum
|
ユーザー |
|
提出日時 | 2021-04-23 23:06:55 |
言語 | PyPy3 (7.3.15) |
結果 |
RE
|
実行時間 | - |
コード長 | 737 bytes |
コンパイル時間 | 194 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 95,744 KB |
最終ジャッジ日時 | 2024-07-04 08:41:02 |
合計ジャッジ時間 | 3,800 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 1 RE * 26 |
ソースコード
import sys input = sys.stdin.readline MAX = 2*10**5+100 MOD = 10**9+7 fact = [0]*MAX #fact[i]: i! inv = [0]*MAX #inv[i]: iの逆元 finv = [0]*MAX #finv[i]: i!の逆元 fact[0] = 1 fact[1] = 1 finv[0] = 1 finv[1] = 1 inv[1] = 1 for i in range(2, MAX): fact[i] = fact[i-1]*i%MOD inv[i] = MOD-inv[MOD%i]*(MOD//i)%MOD finv[i] = finv[i-1]*inv[i]%MOD def C(n, r): if n<r: return 0 if n<0 or r<0: return 0 return fact[n]*(finv[r]*finv[n-r]%MOD)%MOD def f(n, m): return ((n+2)*C(n+m+2, m)%MOD-m-1)*pow(m+1, MOD-2, MOD)%MOD N, M = map(int, input().split()) A = list(map(int, input().split())) ans = sum(A)%MOD for i in range(N-1): ans += f(N-i-2, M-2)*A[i]%MOD ans %= MOD print(ans)