剑指Offer——1～n整数中1出现的次数

剑指Offer——1～n整数中1出现的次数

题目描述

输入一个整数 n ，求1～n这n个整数的十进制表示中1出现的次数。

例如，输入12，1～12这些整数中包含1 的数字有1、10、11和12，1一共出现了5次。

想法

这个题目应该是比较明显的一个数位dp，只不过我忘的差不多了，贴一个讲的很好的数位dp。https://blog.csdn.net/jk211766/article/details/81474632
这里基本上就是一个数位dp的板子，不需要关心前导0，同时要注意在dfs边界的时候要加的不一定是1。具体看代码。

代码

class Solution {
public:
    int num[50];
    int dp[50][50];
    int dfs(int pos, int sta, bool limit) {
    	// 状态号即为1的个数
        if (pos==-1) return sta;
        if (!limit && dp[pos][sta]!=-1) return dp[pos][sta];
        int up=limit?num[pos]:9;
        int ans=0, nsta;
        for (int i=0; i<=up; ++i) {
            if (i==1) nsta=sta+1;
            else nsta=sta;
            ans+=dfs(pos-1, nsta, limit && (i==num[pos));
        }
        if(!limit) dp[pos][sta] = ans;
        return ans;
    }
    int solve(int x) {
        int pos=0;
        while(x) {
            num[pos++]=x%10;
            x/=10;
        }
        return dfs(pos-1, 0, true);
    }
    int countDigitOne(int n) {
        memset(dp, -1, sizeof(dp));
        return solve(n);
    }
};