forked from doocs/leetcode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathSolution2.cpp
42 lines (39 loc) · 1022 Bytes
/
Solution2.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
class Solution {
public:
int a[11];
int dp[11][1 << 11];
int countSpecialNumbers(int n) {
return f(n);
}
int f(int n) {
memset(dp, -1, sizeof dp);
int len = 0;
while (n) {
a[++len] = n % 10;
n /= 10;
}
return dfs(len, 0, true, true);
}
int dfs(int pos, int mask, bool lead, bool limit) {
if (pos <= 0) {
return lead ? 0 : 1;
}
if (!lead && !limit && dp[pos][mask] != -1) {
return dp[pos][mask];
}
int up = limit ? a[pos] : 9;
int ans = 0;
for (int i = 0; i <= up; ++i) {
if ((mask >> i) & 1) continue;
if (i == 0 && lead) {
ans += dfs(pos - 1, mask, lead, limit && i == up);
} else {
ans += dfs(pos - 1, mask | 1 << i, false, limit && i == up);
}
}
if (!lead && !limit) {
dp[pos][mask] = ans;
}
return ans;
}
};