题目很明显是BFS搜索。
刚开始写的时候,想到两个问题,第一个是如何保证这一次改变的数字跟上一次改变的数字不是同一位。
解决方法是在struct里面记录就行了
还有一个问题是是用int保存这个四位数还是用string来保存比较好,大概想一下,用int会比较方便,但是修改数字可能会比较麻烦,后来又想到可以利用之前看过的康拓展开里用到的一个方法,即用一个循环把四位数的int存到一个数组里面,这样去修改就比较方便了。最后就选择了int来保存
第一次写完,基本没有bug,就是在修改最高位的时候,循环里面如果是0,这样就退化成三位数,这个if判断遗漏了。其它都没错。
但是提交还是TLE。于是搜了一下,有两个改进的地方:
一个是可以先求素数表(没有用这个的习惯啊!!)。另一个是比较重要的,可能是很久没写算法的程序了,忘记记录访问过的位置了。加一个isVisited[]就搞定了。
#include <iostream>
#include <queue>
#include <string.h>
bool isPrime[10001];
bool isVisited[10001];
struct node {
int num;
int height;
int last;
node(int n, int h, int l) {
num = n; height = h; last = l;
}
};
using namespace std;
bool judge(int n) {
for (int i = 2; i * i <= n; i++)
if (n % i == 0)
return false;
return true;
}
void f(node& n, int w, queue<node>& q) {
int i = 1, num = n.num, digits[5];
while (num != 0) {
digits[i++] = num % 10;
num /= 10;
}
int flag;
for (i = 0; i <= 9; i++) {
if (i == 0 && w == 4)
continue;
if (i != digits[w]) {
flag = digits[w]; digits[w] = i;
int temp = digits[1] + digits[2] * 10 + digits[3] * 100 + digits[4] * 1000;
if (isPrime[temp] && !isVisited[temp])
q.push(node(temp, n.height + 1, w));
digits[w] = flag;
}
}
}
int f(int start, int end) {
queue<node> q;
q.push(node(start, 0, 0));
while (!q.empty()) {
node temp = q.front(); q.pop();
isVisited[temp.num] = true;
if (temp.num == end)
return temp.height;
for (int i = 1; i <= 4; i++) {
if (i != temp.last)
f(temp, i, q);
}
}
return -1;
}
void makingPrime() {
memset(isPrime, true, sizeof(isPrime));
isPrime[1] = false;
for (int i = 2; i * i <= 10000; i++)
if (isPrime[i]) {
for (int j = 2; j * i <= 10000; j++)
isPrime[i * j] = false;
}
}
int main() {
makingPrime();
int n;
cin >> n;
while (n-- > 0) {
memset(isVisited, false, sizeof(isVisited));
int start, end;
cin >> start >> end;
if (start == end) {
cout << "0" << endl;
continue;
}
int result = f(start, end);
if (result == -1)
cout << "Impossible" << endl;
else
cout << result << endl;
}
return 0;
}