POJ 3126 BFS

题目链接：

POJ 3126 Prime Path

题目描述：

第一行：n
第二到n+1行：两个四位素数，求从第一个数变为第二个数所需的最小步数，每步可将第一个数的一个位(共四位)变为另一个数，且新得到的四位数也为一个素数。

Sample Input

3
1033 8179
1373 8017
1033 1033

Sample Output

6
7
0

思路：

求最短路径问题，bfs，先将1000~10000间素数打表。

伪代码：

//取队首for(每一位):    for(0~9）：        if(是素数　&& 之前没出现过)            //入队，访问，步数＋１；        if(变化后与终点相等) return //步数

完整代码:

/*************************************************************************    > File Name: poj_3126.cpp    > Author:     > Mail:      > Created Time: ************************************************************************/#include<iostream>#include<stdio.h>#include<cstring>#include<cstdlib>#include<algorithm>#include<queue>#define LL long longusing namespace std;const int N = 10000;int n;bool prim[10009];int ans = 0;bool check(int k){    if(k == 0 || k == 1) return 0;    if(k == 3 || k == 2) return 1;    for(int i = 2; i*i <= k; i++)    {        if(k % i == 0) return 0;    }    return 1;}void init(){    for(int i = 1000; i < N; i++)    {        if(check(i)) prim[i] = 1;        else prim[i] = 0;    }}int bfs(int first, int last){    queue<int > q;    bool v[N];    int t[5];    int dis[N];    memset(dis, 0, sizeof(dis));    memset(t, 0, sizeof(t));    memset(v, 0, sizeof(v));    v[first] = 1;    q.push(first);    while(!q.empty())    {        int x = q.front();        if(x == last) return dis[x];        q.pop();        t[0] = x / 1000;        t[1] = x % 1000 / 100;        t[2] = x % 100 / 10;        t[3] = x % 10;        for(int i = 0; i <= 3; i++)        {            int k = t[i];            for(int j = 0; j <=  9; j++)            {                if(j != t[i])                {                    t[i] = j;                    int a = 1000*t[0]+100*t[1]+10*t[2]+t[3];                    if(!v[a] && prim[a])                    {                        q.push(a);                        dis[a] = dis[x]+1;                        v[a] = 1;                    }                    if(a == last) return dis[a];                }            }            t[i] = k;        }    }    return -1;}int main(){    init();    scanf("%d", &n);    int a, b;    for(int i = 0; i < n; i++)    {        scanf("%d%d", &a, &b);        int ans = bfs(a, b);        if(ans != -1) printf("%d\n", ans);        else printf("Impossible\n");    }    return 0;}