spoj1182Sorted bit squence

数位DP + 二分

#include <iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int n,m,k,f[33][33];
void init()
{
    f[0][0]=1;
    for(int i=1;i<=32;i++)
    {
        f[i][0]=f[i-1][0];
        for(int j=1;j<=i;j++) f[i][j]=f[i-1][j-1]+f[i-1][j];
    }
}
int cal(int x,int k)
{
    int sum=0,tot=0,pos=0;
    int bit[33];
    memset(bit,0,sizeof(bit));
    while(x)
    {
        bit[++pos]=x%2;
        x/=2;
    }
    for(int i=pos;i>=1;i--)
    {
        if(!bit[i]) continue;
        sum+=f[i-1][k-tot];
        tot++;
        if(tot>=k) break;
    }
    if(tot==k) sum++;
    return sum;
}
int fnd(int l,int r,int k)
{
    int i,num;
    for(i=1;i<=31;i++)
    {
        num=cal(r,i)-cal(l-1,i);
        if(num>=k) break;
        else k-=num;
    }
    int tmp=l;
    while(l<=r)
    {
        int mid=l+(r-l)/2;
        num=cal(mid,i)-cal(tmp-1,i);
        if(num>=k) r=mid-1;
        else l=mid+1;
    }
    return l;
}
int main()
{
   // freopen("cin.txt","r",stdin);
    int t;
    init();
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d%d",&m,&n,&k);
        if(m<0)
        {
            m&=~(1<<31);
            if(n==0)
            {
                n--;
                k--;
            }
            n&=~(1<<31);
           // printf("m=%d n=%d
",m,n);
            printf("%d
",(1<<31)|fnd(m,n,k));
        }
        else
        {
            if(m==0) k--,m++;
            printf("%d
",fnd(m,n,k));
        }
    }
    return 0;
}