Prime Path-BFS

POJ 3126 Prime Path 题解

最新推荐文章于 2021-05-08 20:01:41 发布

原创最新推荐文章于 2021-05-08 20:01:41 发布 · 165 阅读

0 ·

CC 4.0 BY-SA版权

BFS 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了 POJ 3126 Prime Path 的解题思路及实现过程，采用广度优先搜索（BFS）结合素数判断的方法，在四位数的素数中寻找从一个数到另一个数的最短路径。

3126 – Prime Path http://poj.org/problem?id=3126
题目很简单，但是我特么老是出一堆奇奇怪怪的问题，还有，那些位数计算的错误也是把我坑得有点惨，写这题的时候先把位数计算的过程写在纸上比较不容易错。。。老是犯一些低级错误纠错一整天。。。
阿西吧
在这里插入图片描述

代码不难理解，普通BFS加个素数筛选而已，对每一位进行枚举，判断是否为素数就可以了。。。
为什么纠错纠了两个小时，因为我把文件命名成中文，所以跳出了一个奇怪的东西？
就这个std::ios_base::Init::Init() 这是个啥啊。。。然后还一直输出0.

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<iostream>
#include<cmath>
using namespace std;
int n,m,a,b,Star,final,ans;
int vis[10005];
struct node{
	int step;
	int x;
};
node Now,Next;
queue<node>q;
bool isPrime(int idx){
	
	for(int i=2;i<=sqrt(idx);i++){
		if(idx%i==0)return false;
	}
	return true;
}

void BFS(int idx){
	Now.step=0;
	Now.x=idx;
	q.push(Now);
	vis[idx]=1;
	while(!q.empty()){
		Now=q.front();q.pop();
		if(Now.x==m){
			ans=Now.step;
			return;
		}
		for(int k=0;k<4;k++){
			if(k==0){//千位 
				for(int i=1;i<=9;i++){
					Next=Now;
					Next.x=i*1000+Now.x%1000;
					if(isPrime(Next.x)&&Next.x!=Now.x){
						if(!vis[Next.x]){
							vis[Next.x]=1;
							Next.step++;
							q.push(Next);
						}
					}
				}
			}
			else if(k==1){//百位 
				for(int i=0;i<=9;i++){
					Next=Now;
					Next.x=Now.x/1000*1000+i*100+Now.x%100;
					if(isPrime(Next.x)&&Next.x!=Now.x){
						if(!vis[Next.x]){
							vis[Next.x]=1;
							Next.step++;
							q.push(Next);
						}
					}
				}
			}
			else if(k==2){//十位 
				for(int i=0;i<=9;i++){
					Next=Now;
					Next.x=Now.x/100*100+i*10+Now.x%10;
					if(isPrime(Next.x)&&Next.x!=Now.x){
						if(!vis[Next.x]){
							vis[Next.x]=1;
							Next.step++;
							q.push(Next);
						}
					}
				}
			}
			else{//个位 
				for(int i=0;i<=9;i++){
					Next=Now;
					Next.x=Now.x/10*10+i;
					if(isPrime(Next.x)&&Next.x!=Now.x){
						if(!vis[Next.x]){
							vis[Next.x]=1;
							Next.step++;
							q.push(Next);
						}
					}
				}
			}
		}
	}
	
}
int main()
{	
	int t;
	cin>>t;
	while(t--){
		while(!q.empty())q.pop();
		memset(vis,0,sizeof(vis));
		cin>>n>>m; 
		
		BFS(n);
		cout<<ans<<endl;
	}
	return 0;
}