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