loj#10163Amount of Degrees

版权声明：欢迎转载（标记出处），写得不好还请多指教 https://blog.csdn.net/quan_tum/article/details/82148580

求区间转化为$b$进制$1$的个数为$k$的数的个数。
主要思路就是依靠二进制来做，可以转化为一颗二叉树，这里写得很详细。
先预处理高度为$i$的完全二叉树内二进制表示中恰好含有$j$个$1$的数的个数。这很容易用递推求出：设$f[i,j]$表示所求，则统计其左右子树内符合条件数的个数，即$f[i,j]=f[i-1,j]+f[i-1,j-1]$。
计算时先转换为$b$进制，找到$n$的左起第一位非$0、1$的数位，将它与右面所有数位设为$1$,就可以把这个数视为二进制来做。

#include<bits/stdc++.h>
using namespace std;
int x,y,k,b,l,ans,cnt,a[32],f[35][35];
void init(){
    f[0][0]=1;
    for(int i=1;i<=32;++i){
        f[i][0]=f[i-1][0];
        for(int j=1;j<=i;++j)
        f[i][j]=f[i-1][j]+f[i-1][j-1];
    }
}
int calc(int n){
    l=ans=cnt=0;
    while(n) a[++l]=n%b,n/=b;
    for(int i=l;i;--i){
        if(a[i]==1){
            ans+=f[i-1][k-cnt];
            if(++cnt==k) break;
        }
        else if(a[i]>1){
            ans+=f[i][k-cnt];
            break;
        }
    }
    if(cnt==k) ++ans;
    return ans;
}
int main(){
    init();
    scanf("%d%d%d%d",&x,&y,&k,&b);
    return !printf("%d",calc(y)-calc(x-1));
}