计算各个位数不同的数字个数（Leetcode-357）-回溯法/动态规划

题目

• 原题连接：https://leetcode-cn.com/problems/count-numbers-with-unique-digits/
  

知识点

• 动态规划（回溯超时 ）

思路

• 对于回溯法，枚举每一个位，用一个数组vis记录访问过的数字即可。唯一需要注意的是，这道题允许001（就是1）这种情况的出现，n位以内都是允许的，回溯法代码当n==8会超时，仅供参考。
• 动态规划思路（去计算不能满足条件的数目），首先给出状态转移方程再解释：
  • 对于长度为n的数不能满足条件的数目 = 长度为（n-1）的数不能满足的数目*10 + 长度为（n-1）的数满足的数目 * （n-1）
  • 对于长度为n-1的数不能满足的数目*10：前面n-1位出现重复，那么最后一位是0~9的10个数字都可以
  • 长度为（n-1）的数满足的数目 * （n-1）：前面n-1位没有出现重复，那么后一位从中前面已出现过的数字中选择一个就是不满足的情况，所以总数的话就是n-1位满足的数目*(n-1)
  • 综上，Line 8就是我们的状态转移方程，(9 * pow(10, n - 2) - num_except) 得到的是前n-1位的总数（包含不满足的情况），第一个9是因为首位只能是0-9，所以后面再乘n-2位的总数，最后减去num_except（Line 7求得的长度为（n-1）的数不能满足的数目）

代码

回溯法代码片段

int res = 0;

void dfs(const int &ed, int depth, vector<bool> vis) {
    if (depth == ed) {
        res++;
        return;
    }
    for (int i = 0; i <= 9; i++) {
        if (!vis[i]) {
            vis[i] = true;
            dfs(ed, depth + 1, vis);
            vis[i] = false;
        }
    }
}

int countNumbersWithUniqueDigits(int n) {
    for (int i = 1; i <= n; i++) {
        vector<bool> vis(10, false);
        for (int start = 1; start <= 9; start++) {
            vis[i] = true;
            dfs(i, 1, vis);
            vis[i] = false;
        }
    }
    return res + 1;
}

DP完整代码

class Solution {
public:
    int m[9];

    int dp(int n) {
        if (n == 0 || n == 1) return 0;
        int num_except = dp(n - 1);
        int ans = num_except * 10 + (9 * pow(10, n - 2) - num_except) * (n - 1);
        m[n] = ans;
        return ans;
    }

    int countNumbersWithUniqueDigits(int n) {
        memset(m, 0, sizeof(int[9]));
        dp(n);
        int sum = 0;
        for (int i:m)   sum += i;
        return pow(10, n) - sum;
    }
};