SPOJ SORTBIT - Sorted bit squence(数位统计)

引自：http://wenku.baidu.com/view/d2414ffe04a1b0717fd5dda8.html

题目大意：

    将区间[m,n]内的所有整数按照其二进制表示中 1 的数量从小到大排序。如果 1 的数量相同，则按照数的大小排序。求这个序列中的第 k 个数。其中，负数使用补码来表示：一个负数的二进制表示与其相反数的二进制之和恰好等于 2^32

分析：
    我们首先考虑 m、n 同正的情况。由于排序的第一关键字是 1 的数量， 第二关键字是数的大小， 因此我们很容易确定答案中 1 的个数：依次统计区间[m,n]内二进制表示中含 1 的数量为 0,1,2,…的数(这里需要用到链接文档例一的一段代码)，直到累加的答案超过 k，则当前值就是答案含 1 的个数，假设是 s。利用例一的算法可以解决这个问题。同时， 我们也求出了答案是第几个[m,n]中含 s 个 1 的数。 因此， 只需二分答案，求出[m,ans]中含 s 个 1 的数的个数进行判断即可。
    由于每次询问的复杂度为 O(log(n))，故二分的复杂度为 O(log2(n))，这同时也是预处理的复杂度，因此此算法较为理想。
m<0 的情况， 也不难处理， 我们只要忽略所有数的最高位，求出答案后再将最高位回1 即可。或者也可以直接将负数视为 32 位无符号数，采用同正数一样的处理方法。两种方法都需要特别处理 n=0 的情况。

const int M = 55;
int f[M][M];
void init()
{
        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 cal(int x, int k)//找出区间[0,x]二进制表示中含有k个1的数的个数
{
        int cnt = 0, ans = 0;
        for(int i = 31; i > 0; --i) {
                if(x&(1<<i)) {
                        ++cnt;
                        if(cnt>k) break;
                        x ^= (1<<i);
                }
                if((1<<(i-1))<=x)
                        ans += f[i-1][k-cnt];
        }
        if(cnt+x == k) ++ans;
        return ans;
}
int binasearch(int l, int r, int k)
{
        int index = 1, cnt = 0;
        for(int i = 1; i <= 31; ++i) {
                cnt = cal(r,i)-cal(l-1,i);
                if(k <= cnt) {
                        index = i;
                        break;
                }
                k -= cnt;
        }
        int mid, ll = l;
        while(l<r) {
                mid = l + (r-l)/2;
                cnt = cal(mid,index)-cal(ll-1,index);
                if(cnt<k) l = mid+1;
                else r = mid;
        }
        return l;
}
int main()
{
        int x, y, k, t;
        init();
        scanf("%d", &t);
        while(t--) {
                scanf("%d%d%d", &x, &y, &k);
                if(x>=0) {
                        if(k==0) {
                                printf("0\n");
                                continue;
                        }
                        if(x==0) {
                                k -= 1;
                                x += 1;
                        }
                        printf("%d\n", binasearch(x, y, k));
                } else {
                        if(k==0) {
                                printf("0\n");
                                continue;
                        }
                        if(y==0) {
                                k -= 1;
                                y -= 1;
                        }
                        x &= ~(1<<31);
                        y &= ~(1<<31);
                        printf("%d\n", (1<<31)|binasearch(x, y, k));
                }
        }
        return 0;
}