暑假訓練第14天,POJ-3126 ,BFS

 

這一題就是把1000從10000的質數都找出來,每次只改一位數字然後改變後的必須還是素數,找出從這個數到另一個數最少的步數。

你說這人是不是有病,門牌號非要是素數,我真服,真給小明一樣。

然後我的思路是,先打一個素數表從1000到10000所有的素數都存在一個數組裏,然後用廣搜遍歷,每次改變一位數,加入隊列然後,然後用另一個數組存步數,廣搜的好處就是,只要找到終點就是最短的路徑。

可惜我實現不了,看下大佬的代碼

下面上代碼

#include<iostream>
#include<stdio.h>
#include<queue>
#include<math.h>
#include<string.h>
using namespace std;
bool prime[10000];
void init(){//傻瓜版素數打表 
	int i,j;
	for(i=1000;i<=10000;i++)
	{
		for(j=2;j<i;j++)
		if(i%j==0){
			prime[i]=false;
			break;
		}
		if(j==i) prime[i]=true;
	}
}

int bfs(int first,int last){//傳一個開始,一個結束 
	bool dis[10000];
	queue<int> q;//隊列,準備入隊 
	int v,i,j;
	int temp,vtemp;
	int count[10000],t[4];
	memset(dis,false,sizeof(dis));//初始化 
	memset(count,0,sizeof(count));//初始化都爲0 
	q.push(first);//將第一個數放入隊列 
	dis[first]=true;//並作標記 
	
	while(!q.empty())//當隊列不爲空時,這都是老套路 
	{
		v=q.front();//拿出隊首判斷 
		q.pop();//彈出隊首 
		
		t[0]=v/1000;//保存千位 
		t[1]=v%1000/100;//保存百位
		t[2]=v%100/10;//保存十位
		t[3]=v%10;//保存個位
		
		for(j=0;j<4;j++)//遍歷個十百千 
		{
			temp=t[j];//存下這位數 
			for(i=0;i<10;i++)//然後從0到9開始遍歷 
			if(i!=temp)//如果不相等 
			{
				t[j]=i;//賦值 
				vtemp=t[0]*1000+t[1]*100+t[2]*10+t[3];//得到改變後的數 
				if(!dis[vtemp]&&prime[vtemp])//如果沒有遍歷過,並且是素數 
				{
					count[vtemp]=count[v]+1;//步數+1 
					dis[vtemp]=true;//標記一下,避免重複遍歷 
					q.push(vtemp);//將這個數壓入隊列 
				}
				if(vtemp==last)//如果找到這個數 
				return count[vtemp];//返回步數 
			}
			t[j]=temp;//將這個數在變回去 
		} 
		//if(v==last) return count[v];//如果 隊首等於最後你要找的數 
	}
	return -1;
}
int main(){
	int n,a,b,key;
	init();
	scanf("%d",&n);
	while(n--){
		scanf("%d%d",&a,&b);
		key=bfs(a,b);
		if(key!=-1)
		printf("%d\n",key);
		else printf("Impossible");
	}
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章