leetcode 剑指 Offer 43. 1～n 整数中 1 出现的次数

int dp[11][11];
int a[11];

class Solution {
public:
    long long dfs(int eq,int dep,int sum)
    {
        if(!dep)return sum;
        if(~dp[dep][sum]&&!eq)return dp[dep][sum];
        int mx=eq==1?a[dep]:9;
        long long ret=0;
        for(int i=0;i<=mx;i++)
        {
            ret+=dfs((i==mx)&&eq,dep-1,(sum+((i==1)?1:0)));
        }
        if(!eq)dp[dep][sum]=ret;
        return ret;
    }
    long long work(int n)
    {
        int tot=0;
        memset(dp,-1,sizeof(dp));
        for(;n;n/=10)a[++tot]=n%10;
        return dfs(1,tot,0);
    }
    int countDigitOne(int n) {
        return work(n);
    }
};