Codeforces Round #359 (Div. 2) C. Robbers' watch DFS

C. Robbers' watch

链接：

http://codeforces.com/contest/686/problem/C

题意:

给你n和m，找出(a,b)的对数，其中a满足要求：0<=a<n,a的7进制的位数和n-1的7进制的位数相同，b满足要求：0<=b<m,b的7进制的位数和m-1的7进制的位数相同，且a和b的7进制下的位上的数都不相同，即如果a的七进制数为10，b的7进制数为21，这种情况是不行的，因为重复了1

题解：

先计算n，m各需要多少位（注意7需要1位就够了，而不是两位），枚举每一位的数，一个标记数组就可以了，找完a再找b，注意中间的回溯，找完b检查一下就好了

代码：

#include<iostream>
using namespace std;

int n, m;
int la = 1, lb = 1;
int vis[10];
int ans = 0;

void dfs2(int k,int num)
{
    if (k == lb) {
        ans++;
        return;
    }
    for (int i = 0; i <= 6; i++) {
        if (vis[i] == 0 && num * 7 + i < m) {
            vis[i] = 1;
            dfs2(k + 1, num * 7 + i);
            vis[i] = 0;
        }
    }
}

void dfs1(int k,int num)
{
    if (k == la) {
        dfs2(0, 0);
        return;
    }
    for (int i = 0; i <= 6; i++) {
        if (vis[i] == 0 && num * 7 + i < n) {
            vis[i] = 1;
            dfs1(k + 1, num * 7 + i);
            vis[i] = 0;
        }
    }
}

int main()
{
    cin >> n >> m;
    int fn = n, fm = m;
    if (n > 0)n--;
    if (m > 0)m--;
    while (n / 7) {
        la++;
        n /= 7;
    }
    while (m / 7) {
        lb++;
        m /= 7;
    }
    n = fn, m = fm;
    if (la + lb > 7) {
        cout << 0 << endl;
        return 0;
    }
    dfs1(0, 0);
    cout << ans << endl;
    return 0;
}