CDOJ-#851 方老师与素数(BFS)

题目大意：给定两个四位素数，要求每一次变化只能改变其中一位数字且改变后的数字也必须是素数，问最少需要多少步？

解题思路：直接BFS就能得到，因为两个数都是四位的，先把素数打表标记处理，然后再从第一个数的不同位开始遍历。开始想的是转化为字符串，通过控制下表来处理，结果没弄出来，因为数字只有四位，就直接对每一位进行分别遍历了，详见code。

题目来源：http://www.acm.uestc.edu.cn/#/problem/show/851

code：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
using namespace std;

const int MAXN = 10000;
int T,n,m;
bool prime[MAXN];
bool vis[MAXN];

struct node{
    int num,step;
};

void init_prime(){ //初始化素数表
    prime[1]=1;
    memset(prime,0,sizeof(prime));
    double d=sqrt(10000*1.0);
    for(int i=2; i<=d; i++)
        for(int j=2; j*i<10001; j++)
            prime[i*j]=1;
}

int BFS(){
    node s,t,e;
    queue<node> q;
    s.num=n;
    s.step=0;
    vis[s.num]=1;
    q.push(s);
    while(!q.empty()){
        t=q.front();
        q.pop();
        if(t.num==m) return t.step;
        for(int i=1;i<=9;i++){ //最高位遍历，注意最高位不能为0
            e.num=t.num%1000+i*1000;
            if(!vis[e.num]&&!prime[e.num]){
                vis[e.num]=1;
                e.step=t.step+1;
                q.push(e);
            }
        }
        for(int i=0;i<=9;i++){ //第二位遍历
            e.num=t.num/1000*1000+t.num%100+i*100;
            if(!vis[e.num]&&!prime[e.num]){
                vis[e.num]=1;
                e.step=t.step+1;
                q.push(e);
            }
        }
        for(int i=0;i<=9;i++){ //第三位遍历
            e.num=t.num/100*100+t.num%10+i*10;
            if(!vis[e.num]&&!prime[e.num]){
                vis[e.num]=1;
                e.step=t.step+1;
                q.push(e);
            }
        }
        for(int i=0;i<=9;i++){ //最后一位遍历
            e.num=t.num/10*10+i;
            if(!vis[e.num]&&!prime[e.num]){
                vis[e.num]=1;
                e.step=t.step+1;
                q.push(e);
            }
        }
    }
    return -1;
}

int main(){
    init_prime(); //先打表
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        memset(vis,0,sizeof(vis));
        int ans=BFS();
        if(ans==-1) printf("Impossible\n");
        else printf("%d\n",ans);
    }
    return 0;
}