結果

問題 No.797 Noelちゃんとピラミッド
ユーザー silversmith
提出日時 2019-04-08 07:37:49
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 174 ms / 2,000 ms
コード長 833 bytes
コンパイル時間 73 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 26,648 KB
最終ジャッジ日時 2024-06-28 18:37:13
合計ジャッジ時間 8,377 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 60
権限があれば一括ダウンロードができます

ソースコード

diff # 

N = int(input()) -1
a = list(map(int,input().split()))

def func(N,prime):
    f = [1] * (N + 1)
    for i in range(1,N + 1):
        f[i] = (f[i -1] * i) % prime
    return f

def inv(x,prime):
    _x = x
    inv = 1
    k = prime -2
    while k > 0:
        if k % 2 == 1:
            inv = (inv * _x) % prime
        _x = (_x * _x) % prime   
        k = int(k /2)
    return inv

def func_inv(func,prime):
    N = len(func)-1
    func_inv = [1] * (N+1)
    func_inv[N] = inv(func[N],prime)
    for i in range(1,N):
        func_inv[N - i] = ((N -i + 1) * func_inv[N -i + 1]) % prime
    return func_inv

def comb(N,k, prime):
    return func[N]*func_inv[k]*func_inv[N - k] % prime

prime = 10 ** 9 + 7
func = func(N,prime)
func_inv = func_inv(func,prime)

print(sum([comb(N,k,prime) * a[k] % prime for k in range(N+1)]) % prime)
0