【不太常规的数位 DP】

LeetCode 788. 旋转数字

这题本身数据范围很小，直接暴力没问题，但是很适合来讲数位 DP

题意

• 大致题意：在 1 ~ n 中，满足在所有数位上只出现 {0, 1, 2, 5, 6, 8, 9} 中的数字并且 {2, 5, 6, 9} 至少出现一次的数有多少个。

算法：数位DP

1. 状态定义：定义 $f(i,j,k)$ 为 —— 对于当前数位左边的 i 个数（不包括当前位，即有一个和 n 等长的数字 num ，取其长度为 i 的前缀，允许出现前导零），若这个前缀与 n 的前 i 个数位组成的前缀相同，则 $j=1$，若小于则 $j=0$；若这个前缀中至少出现了 {2, 5, 6, 9} 中的一个，则 $k=1$，反之则 $k=0$。

2. 状态计算：对于当前状态 $f(i,a,b)$，我们需要找到它可以由什么状态转移过来，即找到能转移过来的上一个状态 $f(i+1,x,y)$；在这里，我们先前置一些东西 —— 字符串 s 为 n 转化为字符串的结果；枚举的时候，除了枚举状态中的两个标志 $j$ 和 $k$，还需要枚举当前位填什么数字（由于把 n 转化为字符串，下文均用字母表示）。

• 设当前枚举到的字母为 $c$，若 $c==s[i]$，则对于要找的上一个能转移过来的状态 $f(i+1,x,y)$ 的前缀与 s 的对应前缀是否相等，就只依赖于长度为 i 的前缀，因此 $x=a$，反之，若 $c!=s[i]$ 则长度为 $i+1$ 的前缀一定和 s 的对应前缀不相等，$x=0$；对另一个指标，若 $c$ 是 {2, 5, 6, 9} 中的一个，则一定有 $y=1$，即长度为 $i+1$ 的前缀中一定至少有 {2, 5, 6, 9} 中的一个，反之则 $y=b$，即长度为 $i+1$ 的前缀中是否有 {2, 5, 6, 9} 中的一个，取决于长度为 $i$ 的前缀中是否有。因此状态转移方程为 $f(i,a,b)=f(i,a,b)+f(i+1,x,y)$。

• 初始边界条件：由于这种递推方式是逆着来的，因此初始状态应该设为 $f(m,0,1)=f(m,1,1)=1$（$m$ 为 s 的长度），即对整个串 s，无论枚举到的长度为 $m$ 的前缀与原串是否相同，只要前缀中含有 {2, 5, 6, 9} 中的一个就是合法的一个方案。

• 最终答案：$f(0,1,0)$，表示当前为 s 的首位，前边已经没有前缀，空前缀就视为和枚举的前缀相等，且空前缀不含有 {2, 5, 6, 9} 中的一个。

• 小 $tip$：枚举的时候，如果 $a=1$，即前缀相等，则当前位最多只可枚举到 $s[i]$，否则可以从 $0$ 枚举到 $9$；若枚举到 {3, 4, 7} 中的一个，直接跳过即可。

时间复杂度 $O(logn)$

• 上限为 $36\times logn$，这里是以 $10$ 为底。

C ++ 代码

class Solution {
public:
    int rotatedDigits(int n) {
        string s = to_string(n);
        int m = s.size();
        set<char> s1{'3', '4', '7'}, s2{'2', '5', '6', '9'};

        vector<vector<vector<int>>> f(m + 1, vector<vector<int>>(2, vector<int>(2)));
        f[m][0][1] = f[m][1][1] = 1;

        for (int i = m - 1; i >= 0; i -- )
            for (int a = 0; a < 2; a ++ )
                for (int b = 0; b < 2; b ++ )
                    for (auto c = '0'; c <= (a ? s[i]: '9'); c ++ ) {
                        if (s1.count(c)) continue;
                        int x = c == s[i] ? a: 0;
                        int y = s2.count(c) ? 1: b;
                        f[i][a][b] += f[i + 1][x][y];
                    }

        return f[0][1][0];
    }
};