Acwing 1081 度的数量 (数位dp)

题目链接：点此跳转

题目大意：
求给定区间 [X,Y] 中满足下列条件的整数个数：这个数恰好等于 K 个互不相等的 B 的整数次幂之和。

例如，设 X=15,Y=20,K=2,B=2，则有且仅有下列三个数满足题意：

17=2⁴+2⁰
18=2⁴+2¹
20=2⁴+2²
输入格式
第一行包含两个整数 X 和 Y，接下来两行包含整数 K 和 B。

输出格式
只包含一个整数，表示满足条件的数的个数。

数据范围
1≤X≤Y≤2³¹−1,
1≤K≤20,
2≤B≤10

解题思路：

把num转换成b次幂的数，然后让b次幂的每一位为1的总和为k个即可，这就转换到了数位dp问题，然后再分情况讨论：

对于数字num，存放其每一位上的数字(B进制表示下)，从高位向低位枚举，假设枚举到第i位，第i位上的数字是x，那么分以下几种情况讨论：

1.x == 0

1. 那么第i位只能是0，后面数位上现在都不能确定，只能继续向后看.

2.x == 1
这里第i位可以分成两种情况：

1.第i位放0，那么后面的数位上可以放k-last个1，res += f[i][k-last];
2.第i位放1，那么后面数位上的情况不能用组合数计算，因为要保证答案中的数字比原数字要小

3.x > 1
同样第i位分成两种情况：

1.第i位放0，那么后面的数位上可以放k-last个1，res += f[i][k-last];
2.第i位放1，那么后面的数位上可以放k-last-1个1，res += f[i][k-last-1];

Code:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<algorithm>
#define inf 0x3f3f3f3f

using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn=50;

int k,b;
int f[maxn][maxn];

void init(){    //预处理，求解组合数
	for(int i=0;i<maxn;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];
		}
	}
}

int dp(int n){
	if(!n) return 0; //n为0 直接返回即可
	vector<int> ve;
	while(n) ve.push_back(n%b),n/=b;
	int res=0;
	int last=0; //表示已经取了多少个1
	
	for(int i=ve.size()-1;i>=0;i--){
		int x=ve[i];
		if(x){                    //如果x为0直接考虑下一位
			res+=f[i][k-last];   // 考虑第i位取0的情况，在剩下的（0~i-1位中）取k-last个1
			if(x>1){ //如果x>1，就可以直接用组合数表示出来，不用进行讨论，也就是i位取1的时候，后面i位随便取k-last-1个1
				if(k-last-1>=0) res+=f[i][k-last-1];  
				break;
			}
			else{   //如果x==1，那么i位取1的时候，还要进行讨论，后面i位不能随便取，也就不是组合数
				last++;
				if(last>k) break;
			}
		}
		if(!i&&last==k) res++;
	}
	
	return res;
}



int main(){
	
	init();
	int l,r;
	scanf("%d%d%d%d",&l,&r,&k,&b);
	
	printf("%d\n",dp(r)-dp(l-1));
	
	return 0;
}