动态规划（算法竞赛、蓝桥杯）--数位DP度的数量

1、B站视频链接：E38 数位DP 度的数量_哔哩哔哩_bilibili

#include <bits/stdc++.h> 
using namespace std;
const int N=34;
int a[N];//把B进制数的每一位抠出存入数组 
int f[N][N];//f[i][j]表示在i个位置上，放置j个1的组合数 
int K,B;

void init(){//预处理出组合数
    for(int i=0; i<N; i++) f[i][0]=1;
  for(int i=1; i<N; i++)
    for(int j=1; j<=i; j++)
      f[i][j]=f[i-1][j-1]+f[i-1][j];
}
int dp(int n){
  if(!n) return 0;//特判，n==0返回0 
  int cnt=0; 
  while(n) a[++cnt]=n%B,n/=B;//把B进制数的每一位抠出存入数组 
  
  int res=0, last=0;//last表示第i位之前放置1的个数
  for(int i=cnt; i>=1; i--){//从高位到低位枚举 
    int x=a[i];//取出第i位数字  
    if(x){//第i位==0时，直接跳过，继续枚举下一位 
      res+=f[i-1][K-last];//第i位放0                         
      if(x>1){//第i位>1时         
        if(K-last-1>=0) res+=f[i-1][K-last-1]; //第i位放1  
        break;//第i位放大于1的数，不合要求，则break 
      }
      else{//第i位==1时，不能用组合数计算，继续枚举下一位        
        last++;//更新last        
        if(last>K) break; //1的个数多于K，则break  
      }
    }
      if(i==1 && last==K) res++; //特判，走到末位的情况
  }
  return res;
}
int main(){
  init(); //预处理出组合数 
  int l, r;
  cin>>l>>r>>K>>B;
  cout<<dp(r)-dp(l-1);
  return 0;
}