poj 3126 Prim Path 第一道BFS

poj 3126 Prim Path 第一道BFS

      对于一个四位数，对于它某一位变化之后的素数，即“相邻的素数”，进行广度搜索，知道搜索到为止！
挺简单，看代码应该可以看懂，下面是代码

9 #include  < iostream >
10 #include  < queue >
11 #include  < math.h >
12 using   namespace  std;
13 int  a, b;
14 int  p[ 9999 ]  =   { 0 } ;
15 int  visited[ 9999 ]  =   { 0 } ;
16 bool  isprime( int  x)  {
17
18    for (int i = 2; i <= sqrt((double) x); ++i) {
19        if (x % i == 0)
20            return false;
21    }
22    return true;
23}
24 int  BFS( int  s,  int  r)  {
25    queue<int> q;
26    q.push(s);
27    p[s] = 0;
28    visited[s] = 1;
29    while (!q.empty()) {
30        int temp = q.front();
31        q.pop();
32        for (int i = 0; i <= 9; i++) {
33            int y1 = (temp / 10) * 10 + i;
34            if (isprime(y1) && !visited[y1]) {
35                q.push(y1);
36                p[y1] = p[temp] + 1;
37                visited[y1] = 1;
38            }
39            int y2 = temp % 10 + (temp / 100) * 100 + i * 10;
40            if (isprime(y2) && !visited[y2]) {
41                q.push(y2);
42                p[y2] = p[temp] + 1;
43                visited[y2] = 1;
44            }
45            int y3 = temp % 100 + (temp / 1000) * 1000 + 100 * i;
46            if (isprime(y3) && !visited[y3]) {
47                q.push(y3);
48                p[y3] = p[temp] + 1;
49                visited[y3] = 1;
50            }
51            if (i != 0) {
52                int y4 = temp % 1000 + i * 1000;
53                if (isprime(y4) && !visited[y4]) {
54                    q.push(y4);
55                    p[y4] = p[temp] + 1;
56                    visited[y4] = 1;
57                }
58            }
59            if (visited[r])
60                return p[r];
61        }
62
63    }
64    return 0;
65}
66 int  main()  {
67    int n;
68    cin >> n;
69    while (n--) {
70        memset(visited,0,sizeof(visited));
71        memset(p,0,sizeof(p));
72        cin >> a >> b;
73        cout << BFS(a, b) << endl;
74
75    }
76    return 0;
77}
78