#159302 (C++14) No.498 ワープクリスタル (給料日編)

• 問題: No.498 ワープクリスタル (給料日編)
• ユーザー: pekempey
• 提出日時: 2017-03-24 22:43:52
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: WA
• 実行時間: -
• コード長: 1,165 bytes
• コンパイル時間: 592 ms
• コンパイル使用メモリ: 75,572 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-07-05 22:09:19
• 合計ジャッジ時間: 1,661 ms
• ジャッジサーバーID （参考情報）: judge2 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	WA	-
testcase_05	AC	1 ms; 5,376 KB
testcase_06	WA	-
testcase_07	WA	-
testcase_08	WA	-
testcase_09	AC	25 ms; 5,376 KB
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	WA	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	WA	-
testcase_23	WA	-
testcase_24	WA	-

ソースコード

#include <iostream>
#include <cstdio>
#include <string>
#include <algorithm>
#include <vector>
#include <string>
#include <set>
#include <map>
#include <tuple>

using namespace std;

const int mod = 1e9 + 7;

int fact[1000];
int invfact[1000];
int inv[1000];

int gx, gy, k;
int x[5], y[5], c[5];

int use[5];
int ans;

void dfs(int k, int n) {
	if (k == n) {
		int xx = 0;
		int yy = 0;
		int pattern = 1;
		int s = 0;
		for (int i = 0; i < n; i++) {
			xx += x[i] * use[i];
			yy += y[i] * use[i];
			s += use[i];
			pattern = 1LL * pattern * invfact[use[i]] % mod;
		}
		pattern = 1LL * pattern * fact[s] % mod;
		if (xx == gx && yy == gy) {
			ans += pattern;
			ans %= mod;
		}
		return;
	}
	for (int i = 0; i <= c[k]; i++) {
		use[k] = i;
		dfs(k + 1, n);
	}
}

int main() {
	inv[1] = 1;
	for (int i = 2; i < 1000; i++) {
		inv[i] = 1LL * inv[mod % i] * (mod - mod / i);
	}
	fact[0] = 1;
	invfact[0] = 1;
	for (int i = 1; i < 1000; i++) {
		fact[i] = 1LL * i * fact[i - 1] % mod;
		invfact[i] = 1LL * inv[i] * invfact[i - 1] % mod;
	}

	cin >> gx >> gy >> k;
	for (int i = 0; i < k; i++) {
		cin >> x[i] >> y[i] >> c[i];
	}
	dfs(0, k);
	cout << ans << endl;
}