結果
問題 | No.1081 和の和 |
ユーザー |
![]() |
提出日時 | 2020-06-19 21:38:29 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 31 ms / 2,000 ms |
コード長 | 1,191 bytes |
コンパイル時間 | 84 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 11,008 KB |
最終ジャッジ日時 | 2024-07-03 14:06:47 |
合計ジャッジ時間 | 996 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 8 |
ソースコード
#!/usr/bin/env python3import functoolsM = 10 ** 9 + 7def sum_mod(xs, m=M):return functools.reduce(lambda x, y: (x + y) % m, xs) % mclass ModFact(object):def __init__(self, max_n, p):self.max_n = max_nself.p = pself.fac = [0 for _ in range(max_n)]self.fac[0] = self.fac[1] = 1self.f_inv = [0 for _ in range(max_n)]self.f_inv[0] = self.f_inv[1] = 1self.inv = [0 for _ in range(max_n)]self.inv[1] = 1for i in range(2, max_n):self.fac[i] = self.fac[i - 1] * i % pself.inv[i] = p - self.inv[p % i] * (p // i) % pself.f_inv[i] = self.f_inv[i - 1] * self.inv[i] % pdef binom(self, n, k):if n < k:return 0elif n <0 or k < 0:return 0return self.fac[n] * (self.f_inv[k] * self.f_inv[n - k] % self.p) % self.pdef compute(n, bs):mf = ModFact(n, M)res = sum_mod((bs[k] * mf.binom(n - 1, k) for k in range(n)), M)return resdef main():n = int(input())bs = [int(b) for b in input().split()]res = compute(n, bs)print(res)if __name__ == '__main__':main()