nyoj929 密码宝盒 难

#include <queue> 
#include <string> 
#include <stdio.h> 
#include <string.h> 
#include <algorithm> 
using namespace std; 
const int N = 5005; 
bool vis[N]; 
string ans; 
int a[17],pre[N],ansAt[N],n,m,c; 
//题意有些复杂

int ctoi(char c) 
{ 
    return c >= '0' && c <= '9' ? c - '0' : c - 'A' + 10; 
} 


char itoc(int i) 
{ 
    return i <= 9 ? i + '0' : i + 'A' - 10; 
}


bool bfs() 
{ 
    queue<int> q; 
    q.push(0); 
    while(!q.empty()) 
    { 
        int cur = q.front();
q.pop(); 
        for(int i=0;i<m;i++) 
        { 
            int nxt = (cur * c + a[i]) % n; 
            if(cur == 0 && a[i] == 0 || vis[nxt]) 
            continue; 
            vis[nxt] = true; //标记已经访问过 
            ansAt[nxt] = a[i]; 
            pre[nxt] = cur; 
            
            if(nxt == 0) //C进制且由M个数字中的某些构成的密码是一个给定十进制整数N的正整数倍,目标达成 
            return true; 
            q.push(nxt); 
        } 
    } 
    return false; 
} 
   
bool check() 
{ 
    ans = ""; 
    int p = 0; 
    do{ 
        ans += itoc(ansAt[p]); //数字转换回字符，等待输出 
        p = pre[p]; 
    }while(p); 
    return (int)ans.size() <= 500; 
} 
   
int main() 
{ 
    int T; 
    scanf("%d",&T); 
    while(T--) 
    { 
        memset(vis,false,sizeof(vis)); 
        scanf("%d%d%d",&n,&c,&m); 
        for(int i=0;i<m;i++) 
        { 
            char num[3]; 
            scanf("%s",num); //以免存入空格
            a[i]=ctoi(num[0]); //将字符转换为数字 
        } 
        sort(a,a+m); //排序，方便下面操作 
        
        if(n==0)  //特殊情况，如果该密码是0的整倍数，且排序后第一个数字为0 
        { 
            puts(a[0]==0 ? "0" : "So Sorry."); 
            continue; 
        } 
        if(bfs() && check()) 
        { 
            for(int i=(int)ans.size()-1;i>=0;i--) 
            putchar(ans[i]); 
            puts(""); 
        } 
        else 
        puts("So Sorry."); 
    } 
    return 0; 
}