poj 3126

素数加bfs

代码如下

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;
bool is_prime[10005];
void bfs(int a,int b)
{
    int visit[10005];
    memset(visit,-1,sizeof(visit));
    visit[a]=0;
    queue<int> q;
    q.push(a);
    while(!q.empty()){
        int temp=q.front();
        int s=temp;
        q.pop();
        int num[4],cnt=0;
        while(temp){
            num[cnt++]=temp%10;
            temp/=10;
        }
        for(int i=0;i<=9;i++){
            if(i!=num[0]) {
                temp=num[3]*1000+num[2]*100+num[1]*10+i;
                if(is_prime[temp]&&visit[temp]==-1) {
                    q.push(temp);
                    visit[temp]=visit[s]+1;
                }
            }
        }
        for(int i=0;i<=9;i++){
            if(i!=num[1]) {
                temp=num[3]*1000+num[2]*100+i*10+num[0];
                if(is_prime[temp]&&visit[temp]==-1) {
                    q.push(temp);
                    visit[temp]=visit[s]+1;
                }
            }
        }
        for(int i=0;i<=9;i++){
            if(i!=num[2]) {
                temp=num[3]*1000+i*100+num[1]*10+num[0];
                if(is_prime[temp]&&visit[temp]==-1) {
                    q.push(temp);
                    visit[temp]=visit[s]+1;
                }
            }
        }
        for(int i=1;i<=9;i++){
            if(i!=num[3]) {
                temp=i*1000+num[2]*100+num[1]*10+num[0];
                if(is_prime[temp]&&visit[temp]==-1) {
                    q.push(temp);
                    visit[temp]=visit[s]+1;
                }
            }
        }
        if(visit[b]!=-1) {
            printf("%d\n",visit[b]); break;
        }
    }
}
int main()
{
    for(int i=0;i<=9999;i++)
        is_prime[i]=true;
    for(int i=2;i<=9999;i++){
        if(is_prime[i]) {
            for(int j=2*i;j<=9999;j+=i){
                is_prime[j]=false;
            }
        }
    }
    int t;
    scanf("%d",&t);
    while(t--){
        int a,b;
        scanf("%d%d",&a,&b);
        bfs(a,b);
    }
}