URAL 1057 Amount of Degrees

题目链接：http://acm.timus.ru/problem.aspx?space=1&num=1057

题意：求区间[L,R]内有多少数字将其化成B进制后含有K个1？

思路：首先说二进制，这个比较简单。对于不是二进制的，将其化成B进制后，从高位开始找到第一个大于1的位，将该位以及该位之后的都修改为1.然后将其看做二进制即可。

 

i64 C[N][N];

void init()
{
    C[0][0]=1;
    int i,j;
    for(i=1;i<N;i++)
    {
        C[i][0]=C[i][i]=1;
        for(j=1;j<i;j++) C[i][j]=C[i-1][j-1]+C[i-1][j];
    }
}

int L,R,K,B;


int deal(int n)
{
    int a[N],aNum=0;
    while(n)
    {
        a[aNum++]=n%B;
        n/=B;
    }
    int i;
    for(i=aNum-1;i>=0;i--) if(a[i]>1) break;
    int j;
    for(j=i;j>=0;j--) a[j]=1;
    n=0;
    for(i=aNum-1;i>=0;i--) n=n*2+a[i];
    return n;
}

int cal(int n,int k)
{
    if(B!=2) n=deal(n);
    int ans=0,i,cnt=0;
    for(i=31;i>=1;i--)
    {
        if(n&(1<<i))
        {
            cnt++;
            if(cnt>k) break;
            n^=(1<<i);
        }
        if((1<<(i-1))<=n) ans+=C[i-1][k-cnt];
    }
    if(n+cnt==k) ans++;
    return ans;
}


int main()
{
    init();
    Rush(L)
    {
        RD(R,K,B);
        int ans=cal(R,K)-cal(L-1,K);
        printf("%d\n",ans);
    }
}