POJ3126-Prime Path

题目链接：POJ3126

思路，先打一个素数表，然后BFS的时候，把4位数每一位单独考虑，看看4位数是否为质数，是的话就加入队列。

AC代码：

#include<stdio.h>#include<string.h>#include<queue>using namespace std;const int maxn=10005;bool f[maxn];bool vis[maxn];int n,d1,d2,d3,d4,res;struct node{    int num;    int step;};void tb(){//false is prime    memset(f,false,sizeof(f));    f[0]=true;    f[1]=true;    f[2]=false;    for(int i=2;i<maxn;i++){        if(f[i]) continue;        else{            for(int j=i+i;j<maxn;j+=i){                f[j]=true;            }        }    }}void getnum(int n){    d1=n%10;    d2=(n%100)/10;    d3=(n%1000)/100;    d4=n/1000;}bool bfs(int s,int e){    memset(vis,false,sizeof(vis));    queue<node> q;    node t1,t2,t3;    t1.num=s;    t1.step=0;    q.push(t1);    vis[s]=true;    while(!q.empty()){        t2=q.front();        q.pop();        //printf("%d\n",t2.num);        if(t2.num==e){            res=t2.step;            return true;        }        getnum(t2.num);        //d4        for(int i=1;i<=9;i++){            if(i!=d4){                int tmp=i*1000+d3*100+d2*10+d1;                if(!f[tmp]&&!vis[tmp]){                    t3.num=tmp;                    t3.step=t2.step+1;                    vis[tmp]=true;                    q.push(t3);                }            }        }        //d3        for(int i=0;i<=9;i++){            if(i!=d3){                int tmp=d4*1000+i*100+d2*10+d1;                if(!f[tmp]&&!vis[tmp]){                    t3.num=tmp;                    t3.step=t2.step+1;                    vis[tmp]=true;                    q.push(t3);                }            }        }        //d2        for(int i=0;i<=9;i++){            if(i!=d2){                int tmp=d4*1000+d3*100+i*10+d1;                if(!f[tmp]&&!vis[tmp]){                    t3.num=tmp;                    t3.step=t2.step+1;                    vis[tmp]=true;                    q.push(t3);                }            }        }        //d1        for(int i=0;i<=9;i++){            if(i!=d1){                int tmp=d4*1000+d3*100+d2*10+i;                if(!f[tmp]&&!vis[tmp]){                    t3.num=tmp;                    t3.step=t2.step+1;                    vis[tmp]=true;                    q.push(t3);                }            }        }    }    return false;}int main(){    tb();    scanf("%d",&n);    for(int i=0;i<n;i++){        int a,b;        scanf("%d%d",&a,&b);       // getnum(a);      // printf("%d%d%d%d\n",d4,d3,d2,d1);       // getnum(b);       // printf("%d%d%d%d\n",d4,d3,d2,d1);        if(bfs(a,b)) printf("%d\n",res);        else printf("Impossible\n");    }    return 0;}