国家队论文首题 浅谈数位类统计问题 ural 1057

看《浅谈数位类统计问题》论文写的，感觉分析还是相当的到位啊。

一个数分解成进制数相加，该位置上的数绝对为1.

代码：

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
using std::min;
using std::swap;
int const N = 40;
int f[N][N],num[N],top,x,y,b,k,s,tot;
void pre()
{
     f[0][0]=1;
     for(int i=1;i<=31;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 getsum(int x,int b)
{
    tot=s=top=0;
    while(x)
    {
          num[top++]=x%b;
          x/=b;
    }
    for(int i=top-1;i>=0;i--)
    {
        if(!s&&num[i]>0)s=i;
        if(num[i]>1)
        {
           while(i>=0)x^=(1<<i),i--;
           break;
        }
        if(num[i]==1)
        {
           x^=(1<<i);
        }
    }
    int ret=0;
    s++;
    s=min(31,s);
    for(;s>0;s--)
    {
        if(x&(1<<s))
        {
           tot++;
           if(tot>k)break;
           x^=(1<<s);
        }
        if((1<<(s-1))<=x)
        {
           ret+=f[s-1][k-tot];
        }
    }
    if(tot+x==k)ret++;
    return ret;
}
int main()
{
    pre();
    while(~scanf("%d %d %d %d",&x,&y,&k,&b))
    {
          if(x>y)swap(x,y);
          printf("%d\n",getsum(y,b)-getsum(x-1,b));
    }
    return 0;
}