Prime Path POJ

Prime Path

 POJ - 3126      

比较基础的BFS思想，因为只有四位数，所以把每一位都当做入口，枚举所有可能的数字，遇到符合的就进入队列。

 

#include<iostream>#include <cstdio>#include <string>#include <cstring>#include <algorithm>#include <queue>#include <cmath>using namespace std;int t[10];int n;int prime[10005];int v[10005];int vis[10000];int from,to;int time;void get_prime(){    int x=sqrt(10005)+1;    for(int i=2; i<=10005; i++)    {        if(!prime[i])        {//            cout  << i << endl;            for(int j=i*2; j<10005; j+=i)            {//                cout << j  <<endl;                prime[j]=1;            }        }//        cout<< "2";    }    prime[0]=prime[1]=prime[2]=1;}int bfs(){    queue<int>que;//    que.clear();    que.push(from);    while(!que.empty())    {        int now=que.front();        que.pop();        t[0]=now/1000;        t[1]=(now/100)%10;        t[2]=(now/10)%10;        t[3]=(now%10);        if(now==to) return vis[now];//        printf("%d %d %d %d %d\n",now,t[0],t[1],t[2],t[3]);        for(int i=0; i<4; i++)        {            for(int j=0; j<10; j++)            {                int v;                if(i==0) v=j*1000+t[1]*100+t[2]*10+t[3];                else if(i==1) v=t[0]*1000+j*100+t[2]*10+t[3];                else if(i==2)v=t[0]*1000+t[1]*100+j*10+t[3];                else if(i==3)v=t[0]*1000+t[1]*100+t[2]*10+j;//                cout << v << '*' <<endl;                if(v!=now&& !prime[v]&&!vis[v]&&v>=1000&&v<=9999)                {                    vis[v]=vis[now]+1;                    if(v==to)                    {//                        cout << "*************"<<endl;                        return vis[now]+1;                    }                    que.push(v);//                    cout << "push:" << v <<endl;                }            }        }    }    return -1;}int main(){    get_prime();    scanf("%d",&n);        while(n--)    {        memset(vis,0,sizeof(vis));        scanf("%d%d",&from,&to);        time=bfs();        if(time!=-1)        cout << time <<endl;        else cout << "Impossible\n" ;    }    return 0;}