[AcWing] 338. 计数问题 (C++实现)数位统计dp例题
1. 题目
2. 读题(需要重点注意的东西)
思路: 本题思路不难,主要还是需要分情况讨论清楚,此外,代码实现有边界情况,将思路转换成代码有点复杂。 主要的思路如下: 1、首先计数得分为计0的数 和计除0外的其他数 两种情况; 2、再而,又得针对前三位的取法分为恰好① 取abc 和② 取abc-1 ~ 001(计0时) or000 (计除0外的数)
具体的分类情况如下图所示:
3. 解法
---------------------------------------------------解法---------------------------------------------------
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 10;
int get(vector<int> num, int l, int r)
{
int res = 0;
for (int i = l; i >= r; i -- ) res = res * 10 + num[i];
return res;
}
int power10(int x)
{
int res = 1;
while (x -- ) res *= 10;
return res;
}
int count(int n, int x)
{
if (!n) return 0;
vector<int> num;
while (n)
{
num.push_back(n % 10);
n /= 10;
}
n = num.size();
int res = 0;
for (int i = n - 1 - !x; i >= 0; i -- )
{
if (i < n - 1)
{
res += get(num, n - 1, i + 1) * power10(i);
if (!x) res -= power10(i);
}
if (num[i] == x) res += get(num, i - 1, 0) + 1;
else if (num[i] > x) res += power10(i);
}
return res;
}
int main()
{
int a, b;
while (cin >> a >> b , a)
{
if (a > b) swap(a, b);
for (int i = 0; i <= 9; i ++ )
cout << count(b, i) - count(a - 1, i) << ' ';
cout << endl;
}
return 0;
}
可能存在的问题
4. 可能有帮助的前置习题
5. 所用到的数据结构与算法思想
6. 总结
线性dp的例题,理解思想并自行推导出代码。
|