#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;
}
AcWing 1081. 度的数量 题解(动态规划—DP—数位DP)
最新推荐文章于 2024-04-22 22:21:45 发布