Prime Path------poj3126

 

题目大意：

                给定两个素数a b，求a变幻到b需要几步
                并且变幻时只有一个数字不同，并且是素数

解题思路：

              因为是四位的素数，所以先对素数打表，然后BFS，每次进入队列的是八个只有一个数字不同的四位数，并且还要是素数就入队

#include <cstdio>
#include <iostream>
#include <cstring>
#include <cmath>
#include <queue>
using namespace std;
const int maxn = 1e5;

bool prime[maxn];
//素数打表
void init()
{
    int i, j;
    for(i = 1000; i <= maxn; ++i)
    {
        bool flag = true;
        for(j = 2; j <= sqrt(i); ++j)
            if(i % j == 0)
            {
                prime[i] = false;
                flag = false;
                break;
            }
        if(flag) prime[i] = true;
    }
}
//广搜
int bfs(int first, int last)
{
    bool dis[maxn];
    queue<int> q;
    int count[maxn], t[4];
    memset(dis, false, sizeof(dis));
    memset(count, 0, sizeof(count));

    q.push(first);
    dis[first] = true;

    while(!q.empty())
    {
        int v = q.front();
        q.pop();

        t[0] = v / 1000;
        t[1] = v % 1000 / 100;
        t[2] = v % 100 / 10;
        t[3] = v % 10;

        //改变每一位构造素数
        for(int j = 0; j < 4; ++j)
        {
            //保存要变的数
            int temp = t[j];
            //枚举
            for(int i = 0; i < 10; ++i)
                if(i != temp)
                {
                    t[j] = i;
                    //新的四位数
                    int vt = t[0]*1000+t[1]*100+t[2]*10+t[3];
                    //没有用过且是素数
                    if(!dis[vt] && prime[vt])
                    {
                        //到vt要几步
                        count[vt] = count[v] + 1;
                        //true表示用过
                        dis[vt] = true;
                        //将构造出的素数放进队列
                        q.push(vt);
                    }
                    //如果构造出的数是目标，返回步数
                    if(vt == last) return count[vt];
                }
            //回溯，将改变的那位数变回来
            t[j] = temp;
        }
        if(v == last) return count[v];
    }
    return -1;
}

int main()
{
    int n;
    init();
    cin >> n;
    while(n--)
    {
        int a, b;
        cin >> a >> b;
        int key = bfs(a, b);
        //cout << key << endl;
        if(key != -1) cout << key << endl;
        else cout << "Impossible\n";
    }
    return 0;
}
/*
Sample Input

    3
    1033 8179
    1373 8017
    1033 1033

Sample Output

    6
    7
    0
*/

 

转载于:https://my.oschina.net/Userxs/blog/1618192