leetcode 902 数位dp 不包含0

复习了一下数位dp

肯定不包含0，但是通常数位dp最后计算的结果较小的是包含前导0的，只是没显示出来而已，所以这题需要前导0，但是非前导0是不需要算进去的，因此，加个是否是前导0的状态即可

class Solution {
public:
    int digits[];
    unordered_map<string,int> mp;
    vector<int> dp;
    int dfs(int p,bool e,bool zero){
        if(p==-)   return ;
        if(!e&&!zero&&dp[p]!=-)   return dp[p];
        int u=e?digits[p]:;
        int sum=;
        for(int i=;i<=u;i++){
            if(!mp[string(,i+'')]){
                continue;
            }
            if(i==&&(!zero||(zero&&p==))) continue;
            sum+=dfs(p-,e&&i==u,zero&&i==);
        }
        return e||zero?sum:dp[p]=sum;
    }

    int solve(int n){
        int len=;
        while(n){
            digits[len++]=n%;
            n/=;
        }
        return dfs(len-,,);
    }

    int atMostNGivenDigitSet(vector<string>& D, int N) {
        dp=vector<int>(,-);
        for(auto i : D) mp[i]=;
        mp[string(,'')]=;
        return solve(N);

    }
};