POJ ~ 3126 ~ Prime Path（BFS+打表）

最新推荐文章于 2020-12-18 11:05:41 发布

张松超

最新推荐文章于 2020-12-18 11:05:41 发布

阅读量609

点赞数

分类专栏：【递归/搜索】文章标签： BFS 最短路

本文链接：https://blog.csdn.net/ZscDst/article/details/69651730

版权

【递归/搜索】专栏收录该内容

58 篇文章 0 订阅

订阅专栏

题意：第一个数是几组测试数据，后边每行两个素数，都是四位数，让你从第一个变到第二个数；方式是：每次只能变某一位的数字，而且你每次变成的数字必须为素数；
思路：BFS；先素数打表；bfs过程中如果这个数没出现过，且这个数是素数，且这个数与队头元素只有一位数字不同，就放入队列，然后依次找，统计步数。

//#include<bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
const int MAXN = 1e4 + 5;
bool prime[MAXN];
int s,e;
void found_prime()//筛法素数打表求出10000以内素数
{
    memset(prime, 1, sizeof(prime));
    int m = sqrt(MAXN);
    prime[0] = prime[1] = false;
    for (int i = 2; i < m; i++)
    {
        if (prime[i])
        {
            for (int j = i * i; j < MAXN; j += i)
            {
                prime[j] = false;
            }
        }
    }
}
bool judge(int a, int b)//判断A,B是否只有一位不同
{
    int cnt = 0;
    if (a % 10 != b % 10) cnt++;
    if (a / 10 % 10 != b / 10 % 10) cnt++;
    if (a / 100 % 10 != b / 100 % 10) cnt++;
    if (a / 1000 != b / 1000) cnt++;
    if (cnt == 1) return true;
    return false;
}
int bfs()
{
    bool vis[10000];
    memset(vis, 0, sizeof(vis));
    pair<int, int> NOW;
    NOW.first = s; NOW.second = 0; vis[s] = true;
    queue<pair<int, int> > q;
    q.push(NOW);
    while (!q.empty())
    {
        NOW = q.front();
        q.pop();
        if (NOW.first == e)
        {
            return NOW.second;
        }
        for (int i = 1000; i < 10000; i++)
        {
            if (prime[i] && !vis[i] && judge(i, NOW.first))//i为素数 且没被用过 且只有一位不同
            {
                vis[i] = true;
                q.push(make_pair(i, NOW.second + 1));
            }
        }
    }
    return -1;
}
int main()
{
    found_prime();
    int T;
    scanf("%d", &T);
    while (T--)
    {
        scanf("%d%d", &s, &e);
        int ans = bfs();;
        if(ans != -1) printf("%d\n", ans);
        else printf("Impossible\n");
    }
    return 0;
}
/*
3
1033 8179
1373 8017
1033 1033
*/