题目链接:uva 1548 - The Game of Master-Mind
题目大意:现在ACM公司要开发一个手游,游戏大致为猜数字,一开始给出p,c和m,p为要猜数字的个数,c为每个数字的最大上限,m为已经猜过的次数,每次猜完系统会给出相应的回复提示,所以接下来有2*m行为前m次的提示。每组提示分两行,一行是当前猜的数字分别有哪些,一行是代表说黑点个数和白点个数,所谓黑点个数即当前猜的数字有多少个与答案的位置且颜色均相同(题目中用数字代表颜色),白点则是颜色相同,位置不同的点。要注意的是黑点和白点是一一对应的,例:ans:1 1 2 guess:2 2 1,这种情况下白点个数为2,不是3,就是ans中的2只能对应guess中的一个2,而且如果是黑点的话,就不能算成是白点。现在要求说找出与答案最相近的序列,即满足先前所猜的m次的序列,并且要求字典序最小。
解题思路:dfs,不过光dfs的话,复杂度为100^10。当时因为题目中给的限定条件特别多,所以如果在过程中对黑点白点的情况进行判断,时间上是没有问题的。
但是在判断黑点和白点的时候要注意,统计时要统计黑点个数和点数总和,因为如果将黑点和白点分开计算的话会比较麻烦,因为会有说guess1:2 1,黑1,白0,这样的话ans为1 1是满足的,但是在dfs0层是,ans中的第一个1就变成了白点,如果剪枝为白点个数为0返回就错了。我的处理方法就感觉像是可以先向黑点预支一样,但是预支也有额度,如果超过现存黑点的个数也是不行的。
#include <stdio.h>
#include <string.h>
const int N = 105;
const int M = 15;
int p, c, m, b[N], w[N], ans[M];
int g[N][M], l[N][N], r[N][N];
int cnt[N], sum[N];
void init () {
memset(l, 0, sizeof(l));
memset(r, 0, sizeof(r));
memset(cnt, 0, sizeof(cnt));
memset(sum, 0, sizeof(sum));
scanf("%d%d%d", &p, &c, &m);
for (int i = 0; i < m; i++) {
for (int j = 0; j < p; j++) {
scanf("%d", &g[i][j]);
r[i][g[i][j]]++;
}
scanf("%d%d", &b[i], &w[i]);
}
}
inline bool judge (int x, int d) {
for (int i = 0; i < m; i++) {
if (x == g[i][d]) {
if (b[i] == cnt[i]) return false;
} else if (l[i][x] < r[i][x]) {
if (sum[i] >= w[i] + b[i]) return false;
}
}
return true;
}
inline void set (int x, int d, int flag) {
if (flag > 0) {
for (int i = 0; i < m; i++) {
if (x == g[i][d])
cnt[i]++;
if (l[i][x] < r[i][x])
sum[i]++;
l[i][x]++;
}
} else {
for (int i = 0; i < m; i++) {
if (x == g[i][d])
cnt[i]--;
l[i][x]--;
if (l[i][x] < r[i][x])
sum[i]--;
}
}
}
bool dfs (int d) {
if (d == p) {
for (int i = 0; i < m; i++)
if (cnt[i] != b[i] || sum[i] != w[i] + b[i]) return false;
return true;
}
for (int i = 1; i <= c; i++) {
if (!judge(i, d)) continue;
set(i, d, 1);
ans[d] = i;
if (dfs(d+1)) return true;
set(i, d, -1);
}
return false;
}
int main () {
int cas;
scanf("%d", &cas);
while (cas--) {
init ();
bool flag = dfs(0);
if (flag) {
printf("%d", ans[0]);
for (int i = 1; i < p; i++)
printf(" %d", ans[i]);
printf("\n");
} else
printf("You are cheating!\n");
}
return 0;
}