POJ-3126 Prime Path---BFS+素数打表

题目链接：

https://vjudge.net/problem/POJ-3126

题目大意：

给两个四位数a,b 每次改变a中的一位而且改动之后的必须是素数，问最少改动几次可以到b?(永远达不到b就输出impossible)

思路：

素数打表更好直接判断，然后BFS，用力一点小技巧可以直接生成所有可到达的点

#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<set>
#include<map>
#include<cmath>
using namespace std;
typedef pair<int, int> Pair;
typedef long long ll;
const int INF = 0x3f3f3f3f;
int T, n, m;
const int maxn = 10000 + 10;
bool is_prime[maxn];
void init()
{
    for(int i = 2; i < maxn; i++)is_prime[i] = 1;
    for(int i = 2; i * i < maxn; i++)
    {
        if(is_prime[i])
        {
            for(int j = i * i; j < maxn; j += i)is_prime[j] = 0;
        }
    }
    //for(int i = 0; i < maxn; i++)if(is_prime[i])cout<<i<<endl;
}
bool v[maxn];
void bfs()
{
    int a[4], b[4];
    b[0] = 1000, b[1] = 100, b[2] = 10, b[3] = 1;
    queue<Pair>q;
    memset(v, 0, sizeof(v));
    q.push(Pair(n, 0));
    v[n] = 1;

    while(!q.empty())
    {
        Pair now = q.front();
        q.pop();
        int x = now.first;
        if(x == m)
        {
            cout<<now.second<<endl;
            return;
        }
        a[0] = x % 1000;//每一位置为0
        a[1] = x - x / 100 % 10 * 100;
        a[2] = x - x / 10 % 10 * 10;
        a[3] = x - x % 10;
        for(int i = 0; i < 4; i++)//生成所有的可变化的四位数
        {
            for(int j = 0; j < 10; j++)
            {
                int y = a[i] + j * b[i];
                if(y < 1000 || y == x || !is_prime[y] || v[y])continue;
                v[y] = 1;
                q.push(Pair(y, now.second + 1));
            }
        }
    }
    cout<<"Impossible"<<endl;
    return;
}
int main()
{
    init();
    cin >> T;
    while(T--)
    {
        cin >> n >> m;
        bfs();
    }
}