bfs poj3126 Prime Path

poj3126

开始想了半天没思路，后来百度了下。。。

用bfs对每一位数进行讨论。

要注意的是，千位数不能为零（错了好多次）

也可以用数组存各个位数，利用循环，代码会短很多。

#include<cstdio>
#include<iostream>
#include<queue>
#include<cmath>
#include<cstring>
using namespace std;
struct node
{
int num,step;
};
bool ju[10005];
void is_prime()
{
int i,j;
for(i=1000;i<=9999;i++)
{
for(j=2;j<=sqrt(i);j++)
if(i%j==0)
{
ju[i]=0;
break;
}
if(j>sqrt(i))
ju[i]=1;
}
}
void bfs(int x,int y)
{
node st;
queue<node>q;
st.num=x;
st.step=0;
q.push(st);
ju[st.num]=0;
while(!q.empty())
{
st=q.front();
q.pop();
if(st.num==y)
{
printf("%d\n",st.step);
return;
}
int n1=st.num%10;
int n2=st.num%100/10;
int n3=st.num%1000/100;
int n4=st.num/1000;
node en;
for(int i=1;i<10;i=i+2)
{
en.num=n4*1000+n3*100+n2*10+i;
if(en.num!=st.num&&ju[en.num])
{
en.step=st.step+1;
ju[en.num]=0;
if(en.num==y)
{
printf("%d\n",en.step);
return;
}
q.push(en);
}
}
for(int i=0;i<10;i++)
{
en.num=n4*1000+n3*100+i*10+n1;
if(en.num!=st.num&&ju[en.num])
{
en.step=st.step+1;
ju[en.num]=0;
if(en.num==y)
{
printf("%d\n",en.step);
return;
}
q.push(en);
}
}
for(int i=0;i<10;i++)
{
en.num=n4*1000+i*100+n2*10+n1;
if(en.num!=st.num&&ju[en.num])
{
en.step=st.step+1;
ju[en.num]=0;
if(en.num==y)
{
printf("%d\n",en.step);
return;
}
q.push(en);
}
}
for(int i=1;i<10;i++)
{
en.num=i*1000+n3*100+n2*10+n1;
if(en.num!=st.num&&ju[en.num])
{
en.step=st.step+1;
ju[en.num]=0;
if(en.num==y)
{
printf("%d\n",en.step);
return;
}
q.push(en);
}
}
}
printf("Impossible\n");
return;
}
int main()
{
int n,a,b;

scanf("%d",&n);
while(n--)
{
is_prime();
scanf("%d%d",&a,&b);
bfs(a,b);
}
return 0;
}