ural 1057 Amount of Degrees

题意：统计区间内t=B^n+.....+B^0,每项系数都为0或1，统计1的个数恰好为1的情况，这样的数有多少个

分析：看的刘聪的论文，分析很好，http://pan.baidu.com/s/1pL4FF4n，转化为统计二叉树上1的个数代码写的有点糙

#include<bits/stdc++.h>
using namespace std;
const int maxn=55;
typedef long long ll;
ll f[maxn][maxn],d[maxn];
int b,k;
void init(){
    for(int i=0;i<=31;i++){
        f[i][0]=1;
        for(int j=1;j<=i;j++)
            f[i][j]=f[i-1][j]+f[i-1][j-1];
    }
}

ll cal(ll x){
    int tot=0;ll ret=0;
    for(int i=31;i>0;i--){
        if(x&((ll)1<<i)){
            ++tot;
            if(tot>k)break;
        }
        if(x&(1<<(i-1)))
            ret+=f[i-1][k-tot];
    }
    if(tot+x==k)++ret;
    return ret;
}

ll solve(ll x){
    if(b==2)return cal(x);
    memset(d,0,sizeof(d));
    int len=0;
    while(x>0){
        d[len++]=x%b;
        x/=b;
    }
    int id=len-1;
    while(id>=0&&d[id]<2)id--;
    for(int i=0;i<=id;i++)d[i]=1;
    ll tmp=0,q=1;
    for(int i=0;i<len;i++){
        tmp+=d[i]*q;
        q*=2;
    }
    return cal(tmp);
}

int main(){
    init();ll x,y;
    while(cin>>x>>y>>k>>b)
        cout<<solve(y+1)-solve(x-1)<<endl;

    return 0;
}