POJ - 3126 bfs + 素数筛法 [kuangbin带你飞]专题一

     题意：给定两个四位素数作为终点和起点，每次可以改变起点数的某一位，且改变后的数仍然是素数，问是否可能变换成终点数字？

  思路：bfs搜索，每次改变四位数中的某一位。素数打表方便判断新生成的数是否是素数。

AC代码

#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;
const int maxn = 1e5 + 5;
int vis[maxn], d[maxn];

void deal(int n){
int m = sqrt(n + 0.5);
memset(vis, 0, sizeof(vis));
for(int i = 2; i <= m; ++i) if(!vis[i])
for(int j = i*i; j <= n; j += i) vis[j] = 1;
}

int x, y;

void cut(int n, int *a) {
int ind = 3;
while(n > 0) {
a[ind--] = n % 10;
n /= 10;
}
}

int rev(int *a) {
int n = 0;
for(int i = 0; i < 4; ++i)
n = n * 10 + a[i];
return n;
}

int bfs() {
memset(d, -1, sizeof(d));
d[x] = 0;
queue<int>q;
q.push(x);
while(!q.empty()) {
int n = q.front();
q.pop();
if(n == y) return d[y];
int a[4];
cut(n, a);
for(int i = 0; i < 4; ++i){
int b[4];
memcpy(b, a, sizeof(b));
for(int j = 0; j <= 9; ++j) {

b[i] = j;
int num = rev(b);
if(num > 10000 || num < 1000 || vis[num] || d[num] != -1) continue;
d[num] = d
+ 1;
q.push(num);
}
}
}
return -1;
}

int main(){
deal(maxn);
int T;
scanf("%d", &T);
while(T--) {
scanf("%d%d", &x, &y);
int ans = bfs();
if(ans == -1) printf("Impossible\n");
else printf("%d\n", ans);
}
return 0;
}

如有不当之处欢迎指出！