题目链接:https://www.acwing.com/problem/content/1083/
解题过程:
首先看到这道题目的第一想法是0/1背包。将数字的大小看作是背包的容量,把数字拆分成B进制数字看作是物体的体积,然后需要k个物体。但是这样的时间复杂度是nlogB(n),最极限的情况是 2^32 * 32,会超时。
所以需要使用数位DP的方法;数位DP简而言之就是数字上面的DP,即对每个位置上的数字进行分类讨论,有2种基本的思想:1. f(n):1 - n; 2. 考虑的时候尽量使用树的结构进行讨论。
解题思路如下:
代码:
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int N = 35;
int f[N][N];
int k, b;
inline void init(void) {
for(int i = 0; i <= N; i ++)
for(int j = 0; j <= i; j ++)
if(!j) f[i][j] = 1;
else f[i][j] = f[i - 1][j - 1] + f[i - 1][j];
}
inline int dp(int x) {
if(!x) return 0;
vector<int> v;
while(x) { v.push_back(x % b); x /= b; }
int last = 0;
int res = 0;
for(int i = v.size() - 1; i >= 0; i --) {
x = v[i];
if(x) { //左边的分支
res += f[i][k - last];
if(x > 1) {
if(k - last - 1 >= 0) res += f[i][k - last - 1];
break;
} else {
last ++;
if(last > k) break;
}
}
if(!i && k == last) res ++;
}
return res;
}
int main(void) {
// freopen("in.txt", "r", stdin);
init();
int x, y;
scanf("%d%d%d%d", &x, &y, &k, &b);
printf("%d\n", dp(y) - dp(x - 1));
return 0;
}
被坑了的地方:
这是一个机其傻逼的错误,我将同样的变量在main里面定义了一遍,又在全局变量里面定义了一遍,最后的结果不管怎样都是0;搞得心态有点小崩,最后找了异常的久才找到这个错误点。
得到的收获是,检查的代码的从头到尾检查一遍,首先排除低级错误,再去检查实现错误,最后再检查逻辑错误。