HDU - 1172 - 猜数字（枚举）

#HDU - 1172 - 猜数字
猜数字游戏是gameboy最喜欢的游戏之一。游戏的规则是这样的：计算机随机产生一个四位数，然后玩家猜这个四位数是什么。每猜一个数，计算机都会告诉玩家猜对几个数字，其中有几个数字在正确的位置上。
比如计算机随机产生的数字为1122。如果玩家猜1234,因为1,2这两个数字同时存在于这两个数中，而且1在这两个数中的位置是相同的，所以计算机会告诉玩家猜对了2个数字，其中一个在正确的位置。如果玩家猜1111,那么计算机会告诉他猜对2个数字，有2个在正确的位置。
现在给你一段gameboy与计算机的对话过程，你的任务是根据这段对话确定这个四位数是什么。
Input
输入数据有多组。每组的第一行为一个正整数N(1<=N<=100)，表示在这段对话中共有N次问答。在接下来的N行中，每行三个整数A,B,C。gameboy猜这个四位数为A，然后计算机回答猜对了B个数字，其中C个在正确的位置上。当N=0时，输入数据结束。
Output
每组输入数据对应一行输出。如果根据这段对话能确定这个四位数，则输出这个四位数，若不能，则输出"Not sure"。
Sample Input
6
4815 2 1
5716 1 0
7842 1 0
4901 0 0
8585 3 3
8555 3 2
2
4815 0 0
2999 3 3
0
Sample Output
3585
Not sure
题目链接
这个题目就是说电脑随机给你个数，然后猜数，猜了电脑告诉你有几个数字一样，几个正确的数字位于正确的位置。问你原数是多少，如果能猜得出来，输出数，否则就输出Not sure。

这个题目数据量不算大，可以直接暴力枚举，1000到9999，看看把其中符合条件的数留下，其余去掉。就是在写代码的时候，感觉算有几个数相同时需要注意一下，具体代码里就看懂了。

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 1e5;
int book[maxn];

int cor(int x, int y)
{
    int ans = 0;
    for(int i = 0; i < 4; i++)
    {
        if(x % 10 == y % 10)    ans++;
        x /= 10;
        y /= 10;
    }
    return ans;
}

int cmp(int x, int y)
{
    int a[4], b[4], i = 0, ans = 0;
    while(x)
    {
        a[i++] = x % 10;
        x /= 10;
    }
    i = 0;
    while(y)
    {
        b[i++] = y % 10;
        y /= 10;
    }
    for(i = 0; i < 4; i++)
        for(int j = 0; j < 4; j++)
            if(a[i] == b[j])
            {
                b[j] = -1;
                ans ++;
                break ;
            }
    return ans;
}

int main()
{
    int n;
    while(~scanf("%d", &n), n)
    {
        memset(book, 0, sizeof(book));
        for(int q = 0; q < n; q++)
        {
            int x, m, c;
            scanf("%d%d%d", &x, &m, &c);
            for(int i = 1000; i <= 9999; i++)
            {
                if(book[i]) continue ;
                if(cmp(x, i) != m)  book[i] = 1;
                if(cor(x, i) != c)  book[i] = 1;
            }
        }
        int num = 0, ans;
        for(int i = 1000; i <= 9999; i++)
        {
            if(!book[i])
            {
                num++;
                ans = i;
            }
            if(num >= 2)    break ;
        }
        if(num != 1)    printf("Not sure\n");
        else    printf("%d\n", ans);
    }
    return 0;
}