結果
問題 | No.1127 変形パスカルの三角形 |
ユーザー |
|
提出日時 | 2021-02-28 09:35:33 |
言語 | Kuin (KuinC++ v.2021.9.17) |
結果 |
AC
|
実行時間 | 9 ms / 1,500 ms |
コード長 | 997 bytes |
コンパイル時間 | 2,149 ms |
コンパイル使用メモリ | 148,460 KB |
実行使用メモリ | 6,912 KB |
最終ジャッジ日時 | 2024-09-16 11:40:54 |
合計ジャッジ時間 | 3,583 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
const mod: int :: 1000000007const kMax: int :: 200000var fact: []intvar inv: []intfunc main()var a: int :: cui@inputInt()var b: int :: cui@inputInt()do a :% @moddo b :% @modvar n: int :: cui@inputInt()var k: int :: cui@inputInt()do k :- 1var ans: int :: b * comb(n - 1, k - 1) + a * comb(n - 1, k)do ans :% @moddo cui@print("\{ans}\n")var sum: int :: a ^ 2 + b ^ 2do sum :% @modfor i(1, n - 1)var v: int :: b * comb(n - 1, i - 1) + a * comb(n - 1, i)do v :% @moddo sum :+ v ^ 2do sum :% @modend fordo cui@print("\{sum}\n")func comb(n: int, r: int): intif(@fact =& null)do @fact :: #[@kMax + 1]intdo @inv :: #[@kMax + 1]intdo @fact[0] :: 1for i(1, @kMax)do @fact[i] :: @fact[i - 1] * i % @modend fordo @inv[@kMax] :: math@modPow(@fact[@kMax], @mod - 2, @mod)for i(@kMax, 1, -1)do @inv[i - 1] :: @inv[i] * i % @modend forend ifret @fact[n] * @inv[n - r] % @mod * @inv[r] % @modend funcend func