結果
問題 |
No.797 Noelちゃんとピラミッド
|
ユーザー |
|
提出日時 | 2021-05-22 14:54:12 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 111 ms / 2,000 ms |
コード長 | 539 bytes |
コンパイル時間 | 360 ms |
コンパイル使用メモリ | 82,224 KB |
実行使用メモリ | 111,700 KB |
最終ジャッジ日時 | 2024-10-10 11:34:35 |
合計ジャッジ時間 | 7,119 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 60 |
ソースコード
def cmb(n, r, p): if (r < 0) or (n < r): return 0 r = min(r, n - r) return fact[n] * factinv[r] * factinv[n-r] % p p = 10 ** 9 + 7 N = int(input()) # N は必要分だけ用意する fact = [1, 1] factinv = [1, 1] inv = [0, 1] for i in range(2, N + 1): fact.append((fact[-1] * i) % p) inv.append((-inv[p % i] * (p // i)) % p) factinv.append((factinv[-1] * inv[-1]) % p) A = list(map(int,input().split())) n = N-1 ans = 0 x = 1 for i in range(N): ans += A[i]*x ans %= p x = cmb(n,i+1,p) print(ans)