HDU1226 超级密码(BFS)

由于n最大也只有5000，则对于枚举的密码s，s%n的状态也只有5000种。所以每一个枚举到的状态，开两个内容，一个表示密码的字符串，一个表示模n的余数，当余数为0的时候，说明找到密码。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
#include<cctype>
using namespace std;
struct Node
{
    string s;//表示密码的字符串
    int last;//密码模n的余数
    Node(){}
    Node(string a,int b){s = a; last = b;}
};
string res;
bool num[20],vis[5050],flag;
char str[3];
int a,n,c,m;
void init()
{
    memset(num,0,sizeof num);
    memset(vis,0,sizeof vis);
}
void bfs()
{
    flag = 0;
    queue<Node> Q;
    for(int i = 1; i < 16; i++)//从初始状态开始搜索
        if(num[i])
        {
            Node st;
            if(i <= 9) st.s = i+'0';
            else st.s = i-10+'A';
            st.last = i%n,vis[i%n] = 1,Q.push(st);
        }
    while(!Q.empty())
    {
        Node u = Q.front();
        Q.pop();
        //cout<<u.s<<endl;
        if(u.s.size() > res.size()&&flag) continue;
        if(u.last == 0)//余数为0,找到密码
        {
            if(!flag) res = u.s,flag = 1;
            else if(res.size() > u.s.size()||(res.size() == u.s.size()&&res > u.s)) res = u.s;//找到更短的密码
        }
        for(int i = 0; i < 16; i++)
            if(num[i])
            {
                Node v = u;
                if(i <= 9) v.s += i+'0';
                else v.s += i-10+'A';
                v.last = (v.last*c+i)%n;//c进制数更新密码
                if((v.s.size() <= 500&&!vis[v.last])||!v.last)
                    vis[v.last] = 1,Q.push(v);
            }
    }
    if(flag) cout<<res<<endl;
    else printf("give me the bomb please\n");
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        init();
        res = "";
        scanf("%d%d%d",&n,&c,&m);
        for(int i = 1; i <= m; i++)
        {    
            scanf("%s",str);
            if(isdigit(str[0])) num[str[0]-'0'] = 1;
            else num[str[0]-'A'+10] = 1;
        }
        if(n == 0)
        {
            if(num[0]) printf("0\n");
            else printf("give me the bomb please\n");
            continue;
        }
        bfs();
    }
}