poj 3126 Prime Path（搜索专题）

1 #include <iostream>
2 #include <cstdio>
3 #include <cstring>
4 #include <algorithm>
5 #include <cmath>
6
7 using namespace std;
8
9 bool prime[10000];
10 int que[5000];
11 int step[5000];
12 int isprime[5000];
13 bool vis[10000];
14 int cou=0;
15
16 void getprime(){
17     int num=10000;
18     num=sqrt(num*1.0);
19     memset(prime,true,sizeof(prime));
20     prime[0]=prime[1]=false;
21     for(int i=4;i<10000;i+=2){
22         prime[i]=false;
23     }
24     for(int i=3;i<num;i+=2){
25         if(prime[i]){
26             for(int j=i*i;j<10000;j+=2*i){
27                 prime[j]=false;
28             }
29         }
30     }
31     for(int i=1000;i<10000;i++){
32         if(prime[i]){
33             isprime[cou++]=i;
34         }
35     }
36 }
37
38 bool judge(int a,int b){
39     int sum=0;
40     int t1[4],t2[4];
41     for(int i=0;i<4;i++){
42         t1[i]=a%10;
43         a/=10;
44         t2[i]=b%10;
45         b/=10;
46     }
47     for(int i=0;i<4;i++){
48         if(t1[i]==t2[i]){
49             sum++;
50         }
51     }
52     if(sum==3){
53         return true;
54     }else{
55         return false;
56     }
57 }
58
59 int bfs(int a,int b){
60     int start=0;
61     int endd=0;
62     memset(vis,true,sizeof(vis));
63     que[endd]=a;
64     step[endd++]=0;
65     while(start<endd){
66         int now=que[start];
67         int nowStep=step[start++];
68         if(now==b){
69             return nowStep;
70         }
71         for(int j=0;j<=cou;j++){
72             int i=isprime[j];
73             if(!vis[i]){
74                 continue;
75             }
76             if(prime[i]&&judge(i,now)){
77                 if(i==b){
78                     return nowStep+1;
79                 }
80                 que[endd]=i;
81                 step[endd++]=nowStep+1;
82                 vis[i]=false;
83             }
84         }
85     }
86     return 0;
87 }
88
89 int main()
90 {
91     int n;
92     scanf("%d",&n);
93     int a,b;
94     getprime();
95     for(int i=0;i<n;i++){
96         scanf("%d %d",&a,&b);
97         int ans=bfs(a,b);
98         printf("%d\n",ans);
99     }
100     return 0;
101 }