uva 1548 - The Game of Master-Mind(dfs+剪枝)

题目链接：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;}