代码源：847、素数之欢

本文已参与[新人创作礼]活动，一起开启掘金创作之路

题目描述

这是4月27日代码源div2的每日一题。

素数之欢 - 题目 - Daimayuan Online Judge

现给定两个 四位素数 a,b。 你可以执行多次下面的操作：

修改数字 a 的某一位, 使其成为另一个 四位素数。

例如，1033→1733，其中 1033 与 1733 均为素数。

问至少多少次变换后能从 a 得到 b ? 或回答不可能。

数据规模

多组数据 1≤T≤100

输入格式

第一行一个数字 T，表示接下来将会有 T 组数据。

接下来包含 T 行，每行包含用空格分开的两个 四位素数 a,b。

输出格式

输出 T 行，如果可以，输出最小变换次数。反之输出 −1。

样例输入

2
1033 1033
1033 8779

样例输出

0
5

说明

1033→1733→3733→3739→3779→8779

问题解析

直接暴力bfs搜就可以，题目说了就4位数，我们每次修改4个位置上的数，用0~9替换，替换后判断是否是质数即可，然后计算变换到b一共需要几步。

为了防止t，我们可以先把所有四位数的质数都求出来，这样修改一个位置上的数后，就可以通过直接判断一个数是否在质数集里来判断它是不是质数了，顺便记录一下已经变换过的数，这样下次遇到这个数的时候我们直接不理他即可。

AC代码

#include<iostream>
using namespace std;
#include<vector>
#include<algorithm>
#include<math.h>
#include<set>
#include<numeric>
#include<string>
#include<string.h>
#include<iterator>
#include<map>
#include<unordered_map>
#include<stack>
#include<list>
#include<queue>
#include<iomanip>

#define endl '\n';
typedef long long ll;
typedef pair<ll, ll>PII;
const int N = 2e6 + 50;
int f[100000], a[N];
unordered_map<int, int>mymap;

void get_p()
{
    for (int i = 1000; i <= 9999; i++)
    {
        bool flag = true;
        for (int j = 2; j * j <= i; j++)
        {
            if (i % j == 0)
            {
                flag = false;
                break;
            }
        }
        if (flag)
        {
            mymap[i] = 1;
        }
    }
}

int main()
{
    int t;
    cin >> t;
    get_p();
    while (t--)
    {
        int a, b;
        cin >> a >> b;
        if (a == b)
        {
            cout << 0 << endl;
            continue;
        }
        queue<int>que;
        que.push(a);
        memset(f, 0, sizeof f);
        int res = 1;
        bool flag = false;
        while (!que.empty())
        {
            int len = que.size();
            for (int i = 0; i < len; i++)
            {
                int num = que.front();
                que.pop();
                for (int j = 10; j <= 10000; j*=10)
                {
                    int l = num / j, r = num % (j/10);
                    for (int k = 0; k <= 9; k++)
                    {
                        int ans = l * j + k*(j/10) + r ;
                        if (ans == b)
                        {
                            flag = true;
                            break;
                        }
                        if (f[ans] == 0 && mymap[ans] == 1)
                        {
                            f[ans] = 1;
                            que.push(ans);
                        }
                    }
                    if (flag)break;
                }
                if (flag)break;
            }
            if (flag)break;
            res++;
        }
        cout << res << endl;
    }
    return 0;
}