题面
题解 (数位DP)
题意有点难受,假设K=2,B=2 ,那么要想满足题中要求,在2进制下每一位只能填0或者1,比如 20 = (10100)2,只有这样才能转化为 24 + 2 2 , 再比如K=2,B=8,那么9在8进制下就表示为11, 9=80+81 ,但是17在8进制下表示是21,那么它就不可能表示成为整数次幂之和
所以现在我们就转化成了在[X,Y] 之间的数,在B进制下,由K个1,其余都是0组成的数的个数,数位DP问题我们利用前缀和的思想,统计的是1-N之间满足条件的数,然后在[X,Y] 之间就是 f(Y) - f(X-1)
我们先将数字n转化为B进制的数,将每一位存入数组中,然后从高到低位讨论情况,对于第 i 位数字 x 有3种情况
(1) x = 0 ,那么 i 位只能取0,因为取1就会大于n,直接讨论 i -1 位即可
(2) x =1,则第 i 位可以取0,后面 i-1 位可以随便取,都不会大于n,那么后面就是在 i 位中放剩余1的个数(最多只能放K个),其余都放0
(3) x>1, 那么第 i 位可以取 0,1 ,取0 和 (2) 的情况一样 ,取1 首先K要减去1,然后 在 i 位中放剩余的1的个数,其余放0对于最后一位,K个1都放好,那么就要加上最后一种情况
我们以树的形式考虑
代码
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 35; //2进制下有32位
int K, B;
int f[N][N]; //从i中选j个的方案数
void init() {
for (int i = 0; i < N; i++) {
for (int j = 0; j <= i; j++) {
if (!j) f[i][j] = 1; //从i中选0个的方案数为1
else f[i][j] = f[i - 1][j] + f[i - 1][j - 1];
}
}
}
int dp(int n) {
if (!n) return 0;
vector<int> nums;
//转为B进制
while (n) nums.push_back(n % B), n /= B;
int res = 0;
int last = 0; //已经用了多少个1
for (int i = nums.size() - 1; i >= 0; i--) {
int x = nums[i];
if (x) { //如果是0直接跳过
//左边分支放0
res += f[i][K - last]; //剩余的i位放K-last个1的方案数
if (x > 1) { //左边分支放1
if (K - last - 1 >= 0) res += f[i][K - last - 1];
break; //只能放01,右边分支不合法
} else { //x等于1的情况,看下一位
last++;
if (last > K) break;
}
}
if (!i && K == last) res++;
}
return res;
}
int main() {
init();
int l, r;
cin >> l >> r >> K >> B;
cout << dp(r) - dp(l - 1) << endl;
return 0;
}