数位dp模板（1的个数0

力扣：1的个数
题意：求1-n的所有数中包含的1的个数。
模板：

1. 数组：dp[pos][state] ： 初始化为-1
2. a[len] ： 把数字存储为数组
3. dfs(pos, state, limit)：
   pos代表剩余的可变位数
   state代表前缀中有几个1
   limit代表前缀是否都为0，（是否一直为上界）
4. up = limit ? a[pos] : 9; for循环up，定义tmp累加，!limit 下记忆化到dp数组中

class Solution {
public:
    int a[20];
    // dp[pos][state]代表 剩余pos位可变，当前有state个1时，1的个数（用来记忆化存储）
    int dp[20][20];
    int len;
    //    pos：剩余的可变数位个数， state：前缀中有几个1， limit：前缀是否都为0（上界）
    int digitDfs(int pos, int state, int limit) {
      // pos = 0 代表 已经遍历完一个数的全部位数后，1的个数
      if (pos == -1) {
        return state;
      }
      // 如果dp[][]！=-1，则说明已经搜索过一次，记忆在dp数组里面了
      // !limit是数位dp的条件（直接加）
      if (!limit && dp[pos][state] != -1) {
        return dp[pos][state];
      }
      // limit上界来判断up
      int up = limit ? a[pos] : 9;
      int tmp = 0;
      for (int i = 0; i <= up; ++i) {
        int newstate = state;
        if (i == 1) {
          newstate = state + 1;
        }
        tmp += digitDfs(pos - 1, newstate, limit && i == up);
      }
      // !limit 是数位dp的条件（直接加）
      // 原因是，如果limit时记录，那么是一直是上界，后面位数up选择时不一定能到9
      if (!limit) {
        dp[pos][state] = tmp;
      }
      return tmp;
    }

    int countDigitOne(int n) {
      // 初始化dp
      for (int i = 0; i < 20; ++i) {
        for (int j = 0; j < 20; ++j) {
          dp[i][j] = -1;
        }
      }
      // 把n转化为数组（倒序）
      len = 0;
      while (n) {
        a[len ++] = n % 10;
        n /= 10;
      }
      // 初始化：len-1位可变， 0个前缀1， limit为1
      return digitDfs(len - 1, 0, 1);
    }
};