結果

問題 No.1489 Repeat Cumulative Sum
ユーザー maguroflymagurofly
提出日時 2021-04-23 23:30:07
言語 Ruby
(3.3.0)
結果
TLE  
実行時間 -
コード長 765 bytes
コンパイル時間 210 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 103,740 KB
最終ジャッジ日時 2024-07-04 08:47:35
合計ジャッジ時間 3,967 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 86 ms
17,408 KB
testcase_01 AC 85 ms
12,032 KB
testcase_02 AC 79 ms
12,288 KB
testcase_03 TLE -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
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
# }
class Integer
  def nCr(r); x = 1; (1..r).each { |i| x *= self + 1 - i; x /= i }; x; end
end

# nCk[M + N - k, N - k + 1]
x = (M + N - 1).nCr(N)
# nCk(n, k) = n! / (k! * (n - k)!)
# nCk(n - 1, k - 1) = nCk(n, k) / n / (n - k) * k
ans = (1 .. N).sum { |k| y = x * A[k - 1] % MOD; x = x * (N - k + 1) * (M + N - k).pow(MOD - 2, MOD); y } % MOD
puts ans
0