結果
問題 |
No.1489 Repeat Cumulative Sum
|
ユーザー |
![]() |
提出日時 | 2021-04-23 22:45:55 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 243 ms / 2,000 ms |
コード長 | 653 bytes |
コンパイル時間 | 149 ms |
コンパイル使用メモリ | 82,592 KB |
実行使用メモリ | 91,464 KB |
最終ジャッジ日時 | 2024-07-04 08:28:09 |
合計ジャッジ時間 | 5,358 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
mod = 1000000007 eps = 10**-9 def main(): import sys input = sys.stdin.readline N, M = map(int, input().split()) A = list(map(int, input().split())) N -= 2 M -= 2 fac = [1] * (N+3) finv = [1] * (N+3) for i in range(1, N+3): finv[i] = (finv[i-1] * pow(i, mod-2, mod))%mod fac[i] = (fac[i-1] * (M+i))%mod def comb(i, j): return (fac[i - M] * finv[j])%mod ans = 0 for i, a in enumerate(A): ans += (a * ((((N + 2 - i) * comb(M+(N-i)+2, (N-i)+2))%mod - M - 1) * pow(M+1, mod-2, mod)))%mod ans = (ans + a)%mod print(ans) if __name__ == '__main__': main()