POJ3126 Prime Path 题解&代码

题意：每次给出两个四位素数a和b，要求每次修改a中的一位【要保证修改后仍然是四位数】，询问最少修改几次后能得到b，不能得到的话输出impossible

其实是个bfs水题…结果我判素数浪了一发WA了好久
原因是sqrt(i+0.5)我偷了个懒只写了sqrt(i)结果素数判错了
最后这种水题硬是写了一发对拍拍出来了

#include<iostream>
#include<queue>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
queue<int> q;
int T,a,b,n,tot,flag,vis[10000],prime[10000],pri[10000];
int d[4]={1000,100,10,1};
void init(void)
{
    prime[2]=1;
    pri[tot++]=2;
    for(int i = 3;i < 10000 ;i+=2)
    {
        flag=1;
        for(int j = 0; j < tot && pri[j]<sqrt(i+0.5); j++)//**
            if(i%pri[j]==0)
            {
                flag=0;
                break;
            }
        if(flag)prime[i]=1,pri[tot++]=i;
    }

}
int main(void)
{
    //freopen("test.in","r",stdin);
    //freopen("test.out","w",stdout);
    init();
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&a,&b);
        memset(vis,0,sizeof(vis));
        while(!q.empty())q.pop();
        //if(prime[a])
        //{
            q.push(a);
            vis[a]=1;
        //}
        while(!q.empty())
        {
            int x = q.front();q.pop();
            //printf("%d\n",x);
            if(x==b)break;
            int temp,temp2;
            for(int i = 0; i < 4; i++)
            {
                temp=x;
                temp2=temp/(d[i]*10);temp2*=d[i]*10;temp2+=temp%d[i];
                temp2=max(temp2,1000);
                while(temp-d[i]>=temp2)
                {
                    temp-=d[i];
                    if(!vis[temp] && prime[temp])
                        vis[temp]=vis[x]+1,q.push(temp);
                }
            }
            for(int i = 0; i < 4; i++)
            {
                temp=x;
                temp2=temp/(d[i]*10);temp2*=d[i]*10;temp2+=9*d[i];temp2+=temp%d[i];
                temp2=min(temp2,9999);
                while(temp+d[i]<=temp2)
                {
                    temp+=d[i];
                    if(!vis[temp] && prime[temp])
                        vis[temp]=vis[x]+1,q.push(temp);
                }
            }
        }
        if(!vis[b])printf("Impossible\n");
        else printf("%d\n",vis[b]-1);
    }
    return 0;
}