acwing 1081 度的数量

题面

题解 (数位DP)

题意有点难受，假设K=2,B=2 ,那么要想满足题中要求，在2进制下每一位只能填0或者1，比如 20 = (10100)2,只有这样才能转化为 2⁴ + 2 ² , 再比如K=2，B=8,那么9在8进制下就表示为11， 9=8⁰+8¹ ,但是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;
}