暑假训练第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");
	}
}

猜你喜欢

转载自blog.csdn.net/weixin_42627450/article/details/81195314