POJ 3126 Prime Path

POJ 3126 Prime Path

题目大意：
给出两个四位数的素数a,b，求从a变到b最少要花几步？每一步只能将a中的一位改变，且改变后的数也要是素数。

具体思路：
埃氏筛+BFS
先求出1000-9999间的素数（可用素数筛，非常快捷）不懂的可看这篇博客筛选素数：埃氏筛选、欧拉筛选（线性筛选）、区间筛选
再用bfs搜索，每次变换一个数字，判断是否为质数，并更新变换次数
注意排除前导为0的情况
NC的我以为质数必须是a,b的区间,天知道我贡献了多少次wa

具体代码

#include<iostream>
#include<cmath>
#include<queue>
#include<cstdio>
#include<stdlib.h>
using namespace std;
bool is_prime[10005];
const int INF = 1e5;
int cnt[10005];
int visit[10005];

void init(int a, int b)	//初始化
{
	memset(visit, 0, sizeof(visit));	
	for (int i = 0; i <= 10005; i++)
	{
		cnt[i] = INF;
	}
	cnt[a] = 0;	//到本身的次数为0
}
void sieve(int b)	//埃氏筛
{
	for (int i = 2; i <= b; i++)
		is_prime[i] = true;
	for (int i = 2; i <= b; i++)
		if (is_prime[i])
			for (int j = i * 2; j <= b; j += i)
				is_prime[j] = false;
}

void bfs(int a, int b)
{
	if (a == b)return;
	queue<int> q;
	q.push(a);
	visit[a] = 1;
	while (!q.empty())
	{
		int val = q.front();
		int t = val;
		q.pop();
		int num[4];
		for (int i = 3; i >= 0; i--)
		{
			int left = t % 10;
			num[i] = left;
			t /= 10;
		}
		for (int i = 0; i < 4; i++)	//表示位数
		{
			int temp = num[i];	//暂存
			for (int j = 0; j < 10; j++)	//表示变换数字
			{
				if (i == 0 && j == 0)continue;	//排除掉前导0
				num[i] = j;
				int next = num[0] * 1000 + num[1] * 100 + num[2] * 10 + num[3];	//一步变换得到的数
				if (is_prime[next] && !visit[next] && cnt[next] > 1 + cnt[val])	//求解
				{
					cnt[next] = 1 + cnt[val];
					q.push(next);
					visit[next] = 1;
				}
			}
			num[i] = temp;
		}
	}
}

int main()
{
	int a, b;
	int n;
	scanf("%d", &n);
	sieve(10001);	//先求出10000内的素数
	while (n--)
	{
		scanf("%d%d", &a, &b);
		init(a, b);	//初始化
		bfs(a, b);	//搜索问题的解
		if (cnt[b] != INF)printf("%d\n", cnt[b]);
		else printf("Impossible\n");
	}
	return 0;
}