Prime Path(bfs)

http://poj.org/problem?id=3126

题意：给两个四位数n,m,将n变成m需要多少步，要求每次只能改变n的某一位数，即改变后的数与改变前的数只有一位不同，且每次改变后的数都是素数。

思路：bfs+枚举每一位+素数筛选。

#include <stdio.h>
#include <queue>
#include <string.h>
using namespace std;
const int N=10000;
int prime[N],vis[N],b[4];
void is_prime()
{
    memset(prime,0,sizeof(prime));
    for (int i = 3; i <= 10000; i ++)
    {
        prime[i] = i%2;
    }
    for (int i = 0; i <= 100; i ++)
    {
        if (prime[i])
        {
            for (int j = 2*i; j <= 10000; j += i)
            {
                prime[j] = 0;
            }
        }
    }
    memset(prime,0,1000*sizeof(int));//只保留1000至9999以内的素数
}
int bfs(int l,int r)
{
    int step[N],num ;
    memset(vis,0,sizeof(vis));
    memset(step,0,sizeof(step));
    queue<int>q;
    q.push(l);
    vis[l] = 1;
    while(!q.empty())
    {
        int ll = q.front();
        q.pop();
        b[0] = ll%10;//个
        b[1] = ll/10%10;//十
        b[2] = ll/100%10;//百
        b[3] = ll/1000;//千
        for (int i = 0; i < 4; i ++)
        {
            int t = b[i];//将该位上的数保存，便于恢复
            for (int j = 0; j < 10; j ++)//枚举
            {
                if(t!=j)
                {

                    b[i] = j;
                    num = b[3]*1000+b[2]*100+b[1]*10+b[0];
                    if (!vis[num] && prime[num])//是素数并且没有枚举过
                    {
                        q.push(num);
                        vis[num] =1;
                        step[num] = step[ll]+1;//当前步数为上一状态的步数加1。
                    }
                    if (num==r)
                    {
                        return step[num];
                    }
                }

            }
            b[i] = t;//恢复该位上的数
        }
    }
    return -1;
}
int main()
{
    int t,n,m;
    is_prime();
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        int ans = bfs(n,m);
        if (ans==-1)
            printf("Impossible
");
        else
            printf("%d
",ans);

    }
    return 0;
}