LeetCode 2376. 统计特殊整数

LeetCode 2376. 统计特殊整数

数位DP （灵神模板）

const int N = 10;
class Solution {
public:
    int f[N][1 << N];
    int n;
    string s;
    // u： 当前位， status : 标记每一个数字是否被用过
    // limit : 当前位是否受限制， first : 是否为第一位数字， 不能有前导0 
    int dfs(int u, int status, bool limit, bool first)
    {
        if(u == n) return first == false;
        if(!limit && !first && f[u][status] >= 0) return f[u][status];
        int res = 0;
        if(first) // 是第一位数字，可以跳过
            res += dfs(u + 1, status, false, true);
        for(int i = first, up = limit ? s[u] - '0' : 9; i <= up; i ++)
        {
            if((status >> i & 1) == 0)
                res += dfs(u + 1, status | (1 << i), limit && i == up, false);
        }
        if(!limit && !first) f[u][status] = res;
        return res;
    }
    int countSpecialNumbers(int n) {
        memset(f, -1, sizeof f);
        s = to_string(n);
        this -> n = s.size();
        return dfs(0, 0, true, true);
    }
};