剑指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);
}
};