結果
問題 | No.503 配列コレクション |
ユーザー |
![]() |
提出日時 | 2018-04-17 21:21:29 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 732 bytes |
コンパイル時間 | 563 ms |
コンパイル使用メモリ | 64,404 KB |
実行使用メモリ | 27,004 KB |
最終ジャッジ日時 | 2024-06-27 04:16:16 |
合計ジャッジ時間 | 5,698 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 5 WA * 20 |
ソースコード
#include <iostream>using namespace std;int N, K, D, fact[2000009], inv[2000009], finv[2000009], lim = 2000000, mod = 1000000007;int main() {cin >> N >> K >> D;int P = N / (K - 1), Q = N % (K - 1);if (Q == 0) Q += K - 1, P++;if (D == 1) cout << Q << endl;else {fact[0] = 1, finv[0] = 1;for (int i = 1; i <= lim; i++) {fact[i] = 1LL * fact[i - 1] * i % mod;inv[i] = (i == 1 ? 1 : 1LL * inv[mod % i] * (mod - mod / i) % mod);finv[i] = 1LL * finv[i - 1] * inv[i] % mod;}int ret = 0, pw = 1;for (int i = 0; i <= Q; i++) {ret = (ret + 1LL * pw * fact[P + Q - i - 2] % mod * finv[P - 2] % mod * finv[Q - i]) % mod;pw = 1LL * pw * D % mod;}cout << 1LL * ret * Q << endl;}return 0;}