【SPOJ】1182 Sorted bit sequence

【算法】数位DP

【题解】动态规划

写了预处理函数却忘了调用是一种怎样的体验？

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int f[][];
void init()
{
    f[][]=;
    for(int i=;i<=;i++)
     {
         f[i][]=;
         for(int j=;j<=i;j++)f[i][j]=f[i-][j]+f[i-][j-];
     }
}
int find(int x,int k)
{
    int tot=,ans=;
    for(int i=;i>;i--)
     {
         if((<<i)&x)
          {
              tot++;
              if(tot>k)break;//超过k就没有意义了
              x^=(<<i);
          }
        if((<<(i-))<=x)ans+=f[i-][k-tot];
     }
    if(tot+x==k)ans++;//判断x本身
    return ans;
}
int calc(int l,int r,int k)//统计0...x内二进制表示含k个1的数的个数
{
    int s,now=;
    for(s=;now<k;s++)now+=find(r,s)-find(l-,s);
    s--;now=k-now+find(r,s)-find(l-,s);
    int left=l;
    while(l<r)
     {
         int mid=(int)(((long long)l+(long long)r)>>);//int+int会范围爆炸
         if(find(mid,s)-find(left-,s)>=now)r=mid;else l=mid+;
     }
    return l;
}
int main()
{
    int TT,l,r,k;
    init();
    scanf("%d",&TT);
    while(TT--)
     {
         scanf("%d%d%d",&l,&r,&k);
         if(r<=)
         {
             if(r==)r--,k--;
             l^=(<<),r^=(<<);
             printf("%d\n",(<<)|calc(l,r,k));
             continue;
         }
         if(l==)l++,k--;
         printf("%d\n",calc(l,r,k));
     }
    return ;
}