結果
問題 | 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;}