用BFS写了好久,总是WA, 看了别人的DFS如下:
如我们的n == 21; 那么,我们从个位数1开始搜, 1 * 1 == 1, 9 * 9 == 81,他们的个位数都是1,那么,1和9可以作为我们找的那个数的个位数。
然后我们就记忆一下这个我们找到的东西,放在ans里面。
下面我们开始搜第二位,设搜的是ab的a(其中,我们的b是已经记忆下来的个位)。
ab * ab % 100 = 21; 那么,我们的2 是不是就由(b * b / 10 + a * b * 2) % 10 得到的。于是,我们就for一遍去找a,找到a了,我们就以同样的方法去找c(如果有c的话)。这样,就是剪枝了。
最后,如果我们找的n有x位,那么,我们要找的m * m % ? == n 的m最多也只有x位。至此,就结束了。
代码如下:
#include <cstdio>
#include <iostream>
#include <algorithm>
#define LL long long
#define INF 0x7fffffff
using namespace std;
LL ans, rem, cnt;
int digit[20];
void get_digit(LL n) {
cnt = 0;
while(n) {
digit[++cnt] = n%10;
n /= 10;
}
}
void dfs(LL pos, LL w, LL tmp) {
if(cnt < pos) {
ans = min(ans, tmp);
return ;
}
for(LL k=0; k<10; ++k) {
if((tmp*tmp/w+rem*k%10)%10 == digit[pos])
dfs(pos+1, w*10, k*w+tmp);
}
}
int main(void) {
LL T, n;
cin >> T;
while(T--) {
cin >> n;
ans = INF;
get_digit(n);
for(LL i=0; i<10; ++i) {
if(i*i%10 == digit[1]) {
rem = i<<1;//rem==2b;
dfs(2, 10, i);
}
}
if(ans != INF)
cout << ans << endl;
else cout << "None" << endl;
}
return 0;
}