POJ - 3126 （bfs+最短路）

题目链接
hh学长酷爱素数，他经常自娱自乐，随机挑选两个四位的素数a，b。
游戏规则是：a可以通过改变某一位上的数字使其变成c，但只有当c也是四位的素数时才能进行这种改变。
hh学长可以很轻松的算出a最少经过多少次变化使其变为b，所以他相信你也可以。
例如：1033 -> 8179
1033
1733
3733
3739
3779
8779
8179
最少变换了6次。
Input
第一行输入整数T，表示样例数。 (T <= 100)
每个样例输入两个四位的素数a，b。(没有前导零)
Output
对于每个样例，输出最少变换次数，如果无法变换成b则输出"Impossible"。
Sample Input
3
1033 8179
1373 8017
1033 1033

#include
#include 
#include
#include
#include
#include
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
int top=0,prime[maxn],isprime[maxn],v[maxn],vis[maxn];
void Prime(int n)
{
    for(int i=2;i<=n;++i)
    {
        if(!isprime[i]) prime[top++]=i;
        for(int j=0;j<top;++j)
        {
            if(i*prime[j]>n) break;
            isprime[i*prime[j]]=1;
            if(i%prime[j]==0) break;
        }
    }
}
int bfs(int a,int b)
{
	queue<pair<int,int> >q;
	q.push({a,0});
	v[a]=1;
	while(!q.empty())
	{
		int t=q.front().first,num=q.front().second;
		q.pop();
		if(t==b) return num;
		int t1=t/1000,t2=t/100%10,t3=t/10%10,t4=t%10;
		for(int i=1;i<=9;++i)
		{
			if(i==t1) continue;
			int x=i*1000+t2*100+t3*10+t4;
			if(vis[x]&&!v[x]) q.push({x,num+1}),v[x]=1;
		}
		for(int i=0;i<=9;++i)
		{
			if(i==t2) continue;
			int x=t1*1000+i*100+t3*10+t4;
			if(vis[x]&&!v[x]) q.push({x,num+1}),v[x]=1;
		}
		for(int i=0;i<=9;++i)
		{
			if(i==t3) continue;
			int x=t1*1000+t2*100+i*10+t4;
			if(vis[x]&&!v[x]) q.push({x,num+1}),v[x]=1;
		}
		for(int i=0;i<=9;++i)
		{
			if(i==t4) continue;
			int x=t1*1000+t2*100+t3*10+i;
			if(vis[x]&&!v[x]) q.push({x,num+1}),v[x]=1;
		}
	}
	return -1;
}
int main()
{
	int T,a,b;
 	Prime(1e4);
	for(int i=0;i<top&&prime[i]<1e4;++i)
	if(prime[i]>=1000&&prime[i]<10000) vis[prime[i]]=1;	
	scanf("%d",&T);
	while(T--)
	{
		memset(v,0,sizeof(v));
		scanf("%d %d",&a,&b);
		if(!vis[b]){
			puts("Impossible");continue;
		}
		int ans=bfs(a,b);
		if(ans==-1) puts("Impossible");
		else printf("%d\n",ans);
	}
}