結果

問題 No.1489 Repeat Cumulative Sum
ユーザー magurofly
提出日時 2021-04-23 23:13:17
言語 Ruby
(3.4.1)
結果
TLE  
実行時間 -
コード長 474 bytes
コンパイル時間 454 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 192,404 KB
最終ジャッジ日時 2024-07-04 08:43:03
合計ジャッジ時間 4,942 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other TLE * 1 -- * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

MOD = 10**9 + 7
N, M = gets.split.map(&:to_i)
A = gets.split.map(&:to_i)
A.unshift 0

fac = [1, 1]
fin = [1, 1]
inv = [0, 1]
(2 .. N + M + 10).each do |i|
    inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD
    fac[i] = fac[i - 1] * i % MOD
    fin[i] = fin[i - 1] * inv[i] % MOD
end

nCk = ->(n, k) {
    return 0 if n < k or n < 0 or k < 0
    fac[n] * (fin[k] * fin[n - k] % MOD) % MOD
}

ans = (1 .. N).sum { |k| nCk[M + N - k, N - k + 1] * A[k - 1] % MOD } % MOD
puts ans
0