POJ 3126 Prime Path BFS

题目链接：http://poj.org/problem?id=3126

题意：给定两个四位素数m,n，要求m每次只能变换一位，并且变换前后必须都是质数，同一个数字不能两次访问，求从m变到n的最少次数。

题解：数据都是四位数，很小，可以考虑先打一个所有四位数的素数表，然后BFS。搜索的方法是每次只变换一位，这一位从0到9取。由于不存在前导零，所以千位从1到9变换。

#include <iostream>
#include <cmath>
#include <stdio.h>
#include <algorithm>
#include <string>
#include <cstring>
#include <queue>
#include <map>
#include <set>
using namespace std;

const int maxn = 11000;
bool prime[maxn];
bool vis[maxn];
struct node
{
    int a,b,c,d,st;
    node(){}
    node(int e,int f,int g,int h,int i):a(e),b(f),c(g),d(h),st(i){}
}ss,ns;

void sieve()
{
    memset(prime,1,sizeof(prime));
    prime[0]=prime[1]=0;
    prime[2]=1;
    for(int i=2;i<maxn;i++){
        if(prime[i]){
            for(int j=i*i;j<maxn;j+=i){
                prime[j]=0;
            }
        }
    }
}

int bfs(int n,int m)
{
    memset(vis,0,sizeof(vis));
    if(n==m)    return 0;
    vis[n]=1;
    int a,b,c,d;
    int tmp=n;
    d=tmp%10,tmp/=10;
    c=tmp%10,tmp/=10;
    b=tmp%10,tmp/=10;
    a=tmp%10;tmp/=10;
    queue<node> que;
    que.push(node(a,b,c,d,0));
    while(!que.empty()){
        ss=que.front();que.pop();
        if(ss.a*1000+ss.b*100+ss.c*10+ss.d==m) return ss.st;
        for(int i=0;i<10;i++){
            tmp=ss.a*1000+ss.b*100+ss.c*10+i;
            if(prime[tmp]&&!vis[tmp]&&tmp>1000){
                vis[tmp]=1;
                que.push(node(ss.a,ss.b,ss.c,i,ss.st+1));
            }
        }

        for(int i=0;i<10;i++){
            tmp=ss.a*1000+ss.b*100+i*10+ss.d;
            if(prime[tmp]&&!vis[tmp]&&tmp>1000){
                vis[tmp]=1;
                que.push(node(ss.a,ss.b,i,ss.d,ss.st+1));
            }
        }

        for(int i=0;i<10;i++){
            tmp=ss.a*1000+i*100+ss.c*10+ss.d;
            if(prime[tmp]&&!vis[tmp]&&tmp>1000){
                vis[tmp]=1;
                que.push(node(ss.a,i,ss.c,ss.d,ss.st+1));
            }
        }

        for(int i=1;i<10;i++){
            tmp=i*1000+ss.b*100+ss.c*10+ss.d;
            if(prime[tmp]&&!vis[tmp]&&tmp>1000){
                vis[tmp]=1;
                que.push(node(i,ss.b,ss.c,ss.d,ss.st+1));
            }
        }
    }
    return -1;
}

int main()
{
    sieve();
    int t;
    scanf("%d",&t);
    while(t--){
        int n,m;
        scanf("%d%d",&n,&m);
        int ans=bfs(n,m);
        if(ans!=-1)
            printf("%d\n",ans);
        else
            printf("Impossible\n");
    }
    return 0;
}