pku3126 Prime Path BFS-CSDN博客

题意给定两个四位素数s,e，求出经过最少的步数由s得到e，每一步的要求是每次只能改变四位数中一位数，并且改变后的数要求也要是素数。

首先求出1000到9999所有的素数，然后有s到e进行bfs，知道遇到e结束。

刚开始自己写完代码之后，测试了一下数据，吓死我了，很慢。心想怎么优化呢。想了很长时间，感觉应该能过啊，不耐烦了还是提交了一下。结果1Y。。。囧啊！！不知道威慑呢么机器上运行的很慢。。

看了一下300多秒。

View Code

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#define maxn 10007
#define inf 9999999
using namespace std;

int f[maxn],ans[maxn],b[maxn];
int ct,e,s;

//素数筛选法求出1000 --- 9999 的素数
void init()
{
    int i,j;
    f[0] = f[1] = 1;
    for (i = 2; i*i < maxn; ++i)
    {
        if (!f[i])
        {
            for (j = 2*i; j < maxn; j += i)
            {
                f[j] = 1;
            }
        }
    }
    for (i = 0; i < maxn; ++i)
    {
        if (!f[i] && (i >= 1000 && i <= 9999))
        {
            ans[ct++] = i;
        }
    }
}
//判断是否相差1位
bool isok(int x,int y)
{
    int f1 = 0,f2 = 0,f3 = 0,f4 = 0;
    if (x%10 != y%10) f1 = 1;
    x /= 10; y /= 10;
    if (x%10 != y%10) f2 = 1;
    x /= 10; y /= 10;
    if (x%10 != y%10) f3 = 1;
    x /= 10; y /= 10;
    if (x%10 != y%10) f4 = 1;
    if (f1 + f2 + f3 + f4 == 1)
    return true;
    else
    return false;

}
int bfs()
{
    queue<int>q;
    int i,sum = -1;
    for (i = 0; i < maxn; ++i) b[i] = inf;//记录到达i点的最短距离
    b[s] = 0;
    q.push(s);
    while (!q.empty())
    {
        int cur = q.front(); q.pop();
        if (cur == e)
        {
            sum = b[e];
            break;
        }
        for (i = 0; i < ct; ++i)
        {
            if (isok(cur,ans[i]))
            {
                if (b[ans[i]] > b[cur] + 1)
                {
                     b[ans[i]] = b[cur] + 1;
                     q.push(ans[i]);
                }
            }
        }
    }
    return sum;
}
int main()
{
    int t;
    memset(f,0,sizeof(f));
    init();
    scanf("%d",&t);
    while (t--)
    {
      scanf("%d%d",&s,&e);
      int l = bfs();
      if (l != - 1)
      printf("%d\n",l);
      else
      printf("Impossible\n");
    }
    return 0;
}