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<iostream>
#include<algorithm> 
#include<vector>
#include<cstring>
#include<queue>
#include<cstdio>
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);
	}
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章