POJ 3126 *** Prime Path

题意:给两个1000-9999内的素数A,B，每次改变A的一个位上的数字，得到后的数字必须为素数。问最少需要多少步才能得到B。

想法:对于A而言每次改变一个数字，那么用bfs遍历对A而言所有可能的状态同时判断是否为素数就可以了。

代码如下：

#pragma warning(disable:4996)
#include<iostream>
#include<cstdio>
#include<cmath>
#include<stack>
#include<queue>
#include<cstring>
#include<sstream>
#include<set>
#include<string>
#include<iterator>
#include<vector>
#include<map>
#include<algorithm>
using namespace std;
typedef long long ll;

int prime[10000];
int flag[10000];
int base[4] = { 1,10,100,1000 };

int main(void) {
	//获得素数
	memset(prime, 1, sizeof(prime));
	for (int i = 2; i < 10000; ++i) {
		if (prime[i])
			for (int k = 2; k < 10010 / i; ++k)
				prime[i * k] = 0;
	}

	int n, ori, tar;
	cin >> n;
	while (n--) {
		cin >> ori >> tar;

		//如果ori==tar，直接输出结果
		if (ori == tar) {
			cout << 0 << endl;
			continue;
		}
		
		//如果ori!=tar，按以下方式处理
		memset(flag, 0, sizeof(flag));
		flag[ori] = 1;
		queue<int> path;
		path.push(ori);
		int now, temp, sign = 1;
		while (!path.empty()&&sign) {
			now = path.front();
			path.pop();

			int o[4];
			o[0] = now % 10, o[1] = now % 100 / 10;
			o[2] = now % 1000 / 100, o[3] = now / 1000;

			for (int i = 0; i < 4&&sign; ++i)
				for (int j = 0; j < 10&&sign; ++j) {
					temp = now - o[i] * base[i] + j*base[i];
					if (temp == tar) {
						flag[tar] = flag[now];
						sign = 0;
						break;
					}
					if (!flag[temp] && prime[temp] && temp>1000) {
						flag[temp] = flag[now] + 1;
						path.push(temp);
					}
			}
		}
		if (flag[tar])cout << flag[tar] << endl;
		else cout << "Impossible" << endl;
	}
	return 0;
}