ural 1057 Amount of Degrees(数位统计)

求给定区间[X,Y]中满足下列条件的整数个数:这个数恰好等于 K 个互不相等的 B 的整数次幂之和。(1<=X<=Y<=2^31-1,1<=K<=20,1<=B<=10)

---

思路:
ans=Count(Y)-Count(X-1)
若一个数中有一个B进制的系数大于1->将这之后的系数均置为1

---

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int B,K;
int C[40][40],dig[40];

int dfs(int pos,int limit,int remain){
    if(pos==-1) return remain==0 ? 1:0;
    if(remain<0)    return 0;
    if(!limit)  return C[pos+1][remain];
    int last=(limit==1 ? dig[pos]:1),ans=0;
    for(int i=last;i>=0;i--)
        ans+=dfs(pos-1,limit&(i==last),remain-(i==1 ? 1:0));
    return ans;
}

int solve(int num){
    int len=0;
    while(num){
        dig[len++]=num%B;
        num/=B;
    }
    for(int i=len-1;i>=0;i--)
        if(dig[i]>1){
            for(int j=0;j<=i;j++)
                dig[j]=1;
            break;
        }
    return dfs(len-1,1,K);
}

int main(){
    int L,R;
    C[0][0]=C[1][0]=C[1][1]=1;
    for(int i=2;i<=32;i++)
        for(int j=0;j<=i;j++){
            if(i==j||j==0)   C[i][j]=1;
            else    C[i][j]=C[i-1][j-1]+C[i-1][j];
           // printf("%d %d %d\n",i,j,C[i][j]);
        }
    while(scanf("%d%d%d%d",&L,&R,&K,&B)!=EOF)
        printf("%d\n",solve(R)-solve(L-1));
    return 0;
}