結果
問題 |
No.498 ワープクリスタル (給料日編)
|
ユーザー |
![]() |
提出日時 | 2023-02-22 16:44:51 |
言語 | C++17(clang) (17.0.6 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 43 ms / 2,000 ms |
コード長 | 1,962 bytes |
コンパイル時間 | 1,148 ms |
コンパイル使用メモリ | 142,932 KB |
実行使用メモリ | 7,620 KB |
最終ジャッジ日時 | 2024-07-22 18:17:23 |
合計ジャッジ時間 | 2,545 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 21 |
ソースコード
#include <cassert> #include <cmath> #include <algorithm> #include <iostream> #include <iomanip> #include <climits> #include <map> #include <queue> #include <set> #include <cstring> #include <vector> using namespace std; typedef long long ll; const int MAX_N = 200'010; const ll MOD = 1000000007; ll fac[MAX_N]; ll inv[MAX_N]; ll finv[MAX_N]; class ModInteger { public: ModInteger(int N) { fac[0] = fac[1] = 1; inv[0] = inv[1] = 1; finv[0] = finv[1] = 1; for (int i = 2; i <= N; ++i) { fac[i] = fac[i - 1] * i % MOD; inv[i] = MOD - inv[MOD % i] * (MOD / i) % MOD; finv[i] = finv[i - 1] * inv[i] % MOD; } } ll combination(int n, int k) { if (n < k) return 0LL; if (n < 0 || k < 0) return 0LL; return fac[n] * (finv[k] * finv[n - k] % MOD) % MOD; } ll permutation(int n, int k = -1) { if (k == -1) k = n; if (n < k) return 0LL; if (n < 0 || k < 0) return 0LL; return fac[n] * (finv[n - k] % MOD) % MOD; } ll repeated_combination(int n, int k) { return combination(n + k - 1, k); } }; ModInteger mi(100); int K; int GX, GY; int X[5]; int Y[5]; int N[5]; ll ans; void dfs(int idx, vector<int> counter) { if (idx == K) { int y = 0; int x = 0; int sum = 0; for (int i = 0; i < K; ++i) { int cnt = counter[i]; sum += cnt; y += cnt * Y[i]; x += cnt * X[i]; } if (GY == y && GX == x) { ll val = 1; for (int i = 0; i < K; ++i) { int cnt = counter[i]; val *= mi.combination(sum, cnt); val %= MOD; sum -= cnt; } ans += val; ans %= MOD; } } else { for (int c = 0; c <= N[idx]; ++c) { counter[idx] = c; dfs(idx + 1, counter); } } } int main() { ans = 0; cin >> GX >> GY >> K; for (int i = 0; i < K; ++i) { cin >> X[i] >> Y[i] >> N[i]; } vector<int> counter(K, 0); dfs(0, counter); cout << ans << endl; return 0; }