AcWing 1081. 度的数量 题解（动态规划—DP—数位DP）

原题传送门

#include<bits/stdc++.h>

using namespace std;

const int N = 35;

int X, Y, K, B;
int f[N][N];
vector<int>V;

void init(){
	for(int i = 0; i < N; i ++ ){
		for(int j = 0; j <= i; j ++ ){
			if(!j) f[i][j] = 1;//在i位中选1位的方案数为1 
			else f[i][j] = f[i - 1][j] + f[i - 1][j - 1];//求在i位中选出j位的组合数公式 
		}
	}
}

int dp(int n){//求从1~n中符合条件的数的个数 
	if(!n) return 0;//直接返回方案数为0
	vector<int>V;
	while(n){
		V.push_back(n % B);
		n /= B;
	} 
	int res = 0;//记录个数
	int last = 0;//记录目前有几个符合要求的
	
	for(int i = V.size() - 1; i >= 0; i -- ){
		int x = V[i];
		if(x){
			res += f[i][K - last];//直接加上当x为0时，从i位中选K-last个1的方案数 
			if(x > 1){//如果x>1，则第i位可以选1也可以不选1，不选1的情况与上面为0的情况一样，不讨论，只考虑选1的情况 
				if(K - last - 1 >= 0) res += f[i][K - last - 1];
				break;
			}
			else{
				last ++ ;//如果这一位为1，记录 ++  
				if(last > K) break; //当1的个数大于要求时，退出 
			}
		}
		if(!i && last == K) res ++ ;//考虑最后一位数，此时i=0,如果last == K，则这个方案也符合要求，方案数++ 
	}
	return res;
}

int main()
{
	init();
	
	cin>>X>>Y>>K>>B;
	cout<<dp(Y) - dp(X - 1)<<endl;
	return 0;
}