POJ 3126 Prime Path bfs 简单

给出2个四位数的素数a，b，要求a每次变化只可以变a的4个数字的其中一个，并且变化后的数也要是素数，问a至少要变化多少次才可以变为b。

注意，a的千位数不能变化为0

 

先打出素数表，再bfs数a即可。

 

 

#include<cstdio>
#include<cstring>
#include<queue>

using namespace std;

const int maxn=10010;
const int inf=0x3f3f3f3f;

bool prime[maxn];
bool vis[maxn];

void init_prime()
{
    memset(prime,true,sizeof prime);
    prime[1]=false;
    for(int i=2;i<maxn;i++)
    {
        if(prime[i])
        {
            for(int j=i*2;j<maxn;j+=i)
                prime[j]=false;
        }
    }
    return ;
}
struct Node
{
    int num,cnt;
};

inline int pow(int x,int y)
{
    int ret=1;
    for(int i=1;i<=y;i++)
        ret*=x;
    return ret;
}

int bfs(int a,int b)
{
    if(a==b)
        return 0;
    memset(vis,false,sizeof vis);
    Node str;
    str.num=a;
    str.cnt=0;
    queue<Node>que;
    while(!que.empty())
        que.pop();
    que.push(str);
    vis[str.num]=true;
    while(!que.empty())
    {
        Node u=que.front();
        que.pop();
        if(u.num==b)
            return u.cnt;
        for(int i=1;i<=4;i++)
        {
            for(int j=0;j<10;j++)
            {
                //第i位改为j
                if(i==4&&j==0)
                    continue;
                int x=(u.num%pow(10,i))/pow(10,i-1);
                if(x==j)
                    continue;
                Node tmp;
                tmp.num=u.num-x*pow(10,i-1)+j*pow(10,i-1);
                if(!prime[tmp.num])
                    continue;
                if(vis[tmp.num])
                    continue;
                tmp.cnt=u.cnt+1;
                que.push(tmp);
                vis[tmp.num]=true;
            }
        }
    }
}

int main()
{
    init_prime();
    int test;
    scanf("%d",&test);
    while(test--)
    {
        int a,b;
        scanf("%d%d",&a,&b);

        printf("%d\n",bfs(a,b));
    }
    return 0;
}