POJ - 3126 (Prime Path)

题意：四位素数，若两个素数之间只有一位数不同则它们之间路径为 1，给出不超过100 组四位素数 a 和 b ，求 a 和 b 之间的最短路，若无法到达则输出 "Impossible" ；

 

分析：因为测试数据组数不超过 100 组，所以直接用BFS队列求，先预先筛出 1000~10000 之间的素数，分别标记编号就可以了；

 

代码：

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

typedef pair<int,int> P;
map<int,int>mp;

int prime[1300],cnt;
bool  vis[1300];

bool is_p(int x){
	if(x==2) return 1;
	if(x<2||x%2==0) return 0;
	for(int i=3;i*i<=x;i++){
		if(x%i==0) return 0;
	} 
	return 1;
}

void init(){
	cnt=0;
	for(int i=1001;i<10000;i+=2){
		if(is_p(i)){
			mp[i]=cnt;
			prime[cnt]=i;
			cnt++;
		}
	}
}

int BFS(int x,int y){
	vis[x]=1;
	queue<P>que;
	que.push(P(x,0));
	while(!que.empty()){
		P top=que.front(); que.pop();
		if(top.first==y){
			return top.second;
		}
		int number=prime[top.first];
		int a=number/1000;
		int b=number/100%10;
		int c=number/10%10;
		int d=number%10;
		for(int i=0;i<10;i++){
			if(a!=i&&i!=0){   //不能是前导零
				int num=i*1000+b*100+c*10+d;
				if(is_p(num)&&vis[mp[num]]==0){
					vis[mp[num]]=1;
					que.push(P(mp[num],top.second+1));
				}
			}
			if(b!=i){
				int num=a*1000+i*100+c*10+d;
				if(is_p(num)&&vis[mp[num]]==0){
					vis[mp[num]]=1;
					que.push(P(mp[num],top.second+1));
				}
			}
			if(c!=i){
				int num=a*1000+b*100+i*10+d;
				if(is_p(num)&&vis[mp[num]]==0){
					vis[mp[num]]=1;
					que.push(P(mp[num],top.second+1));
				}
			}
			if(d!=i){
				int num=a*1000+b*100+c*10+i;
				if(is_p(num)&&vis[mp[num]]==0){
					vis[mp[num]]=1;
					que.push(P(mp[num],top.second+1));
				}
			}
		}
	}
	return -1;
}

int main()
{
	init();
	int n;
	scanf("%d",&n);
	while(n--){
		int a,b;
		scanf("%d%d",&a,&b);
		memset(vis,0,sizeof(vis));
		int ans=BFS(mp[a],mp[b]);
		if(ans==-1){
			puts("Impossible");
		}
		else{
			printf("%d\n",ans);
		}
	}
}