[LeetCode]233 Number of Digit One（数位DP）

题目链接：https://leetcode.com/problems/number-of-digit-one/?tab=Description

题意：求[1,n]内1的个数。

裸数位DP，好久不写都快不会写了。。这么水

class Solution {
public:
    int dp[30][30];
    int digit[30];

    int dfs(int l, int cnt, bool flag) {
        if(l == 0) return cnt;
        if(!flag && ~dp[l][cnt]) return dp[l][cnt];
        int pos = flag ? digit[l] : 9;
        int ret = 0;
        for(int i = 0; i <= pos; i++) {
            ret += dfs(l-1, cnt+(i==1), flag&&(pos==i));
        }
        if(!flag) dp[l][cnt] = ret;
        return ret;
    }

    int countDigitOne(int n) {
        memset(dp, -1, sizeof(dp));
        int pos = 0;
        while(n) {
            digit[++pos] = n % 10;
            n /= 10;
        }
        return dfs(pos, 0, true);
    }
};