hdu 4394 Digital Square(搜索：DFS)

用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;
}