POJ - 3126 - Prime Path(BFS枚举)

题目链接

题意：

给出两个四位数的素数a,b，求从a变到b最少要花几步？每一步只能将a中的一位改变，且改变后的数也要是素数。

题解：

BFS枚举个位，十位，百位，千位变换的数，用素数表判断变换后是否为一个素数

AC代码：

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e4+5;
bool prime[maxn];
int vis[maxn];
void Prime(){
    prime[0]=prime[1]=1;
    for(int i=2;i<maxn;i++){
        if(!prime[i])
            for(int j=2;j*i<maxn;j++)prime[i*j]=1;
    }
}
int n,m;
int bfs(){
    int ans=0;
    queue<int>dq;
    dq.push(n);
    vis[n]=1;
    if(n==m)return vis[n];
    while(!dq.empty()){
        int x=dq.front();
        dq.pop();
        if(x==m)return vis[x];
        for(int i=1;i<10;i+=2){ //个位
            int xx=x-x%10+i;
            if(!prime[xx]&&vis[xx]==0){
                dq.push(xx);
                vis[xx]=vis[x]+1;
                if(xx==m)return vis[xx];
            }
        }
        for(int i=0;i<10;i++){  //十位
            int xx=x-(x%100-x%10)+i*10;
            if(!prime[xx]&&vis[xx]==0){
                dq.push(xx);
                vis[xx]=vis[x]+1;
                if(xx==m)return vis[xx];
            }
        }
        for(int i=0;i<10;i++){  //百位
            int xx=x-(x%1000-x%100)+i*100;
            if(!prime[xx]&&vis[xx]==0){
                dq.push(xx);
                vis[xx]=vis[x]+1;
                if(xx==m)return vis[xx];
            }
        }
        for(int i=1;i<10;i++){  //千位
            int xx=x-(x%10000-x%1000)+i*1000;
            if(!prime[xx]&&vis[xx]==0){
                dq.push(xx);
                vis[xx]=vis[x]+1;
                if(xx==m)return vis[xx];
            }
        }
    }
    return -1;
}
int main(){
    Prime();
    int t;
    scanf("%d",&t);
    while(t--){
        memset(vis,0,sizeof(vis));
        scanf("%d%d",&n,&m);
        int ans=bfs();
        if(ans==-1)printf("Impossible\n");
        else printf("%d\n",ans-1);
    }
    return 0;
}