POJ 3252 Round Numbers(组合数学)

题目链接

SB了。。。20亿，我居然用了int,本来觉得这个组合情况比较多，很难写，费了好大劲错了几遍之后。。自己都觉得没错了。。还是WA，找AC代码对拍，1000以下的数据都没问题。。我仔细一看题，这是2后边9个0。。。

题意：求start - end之间多少个二进制时0的个数>=1的个数。

思路：求出1-key的数，相减即可。假如key二进制有len位，那么1到len-1的数都符合。特殊讨论len位的时候的情况。

#include <cstdio>
#include <cstring>
#include <string>
using namespace std;
long long ba[40];
long long c[40][40];
long long judge(long long key)
{
    int ans = 0,i,j,k,t1,t2;
    if(key == 1||key == 0) return 0;
    for(i = 1;i <= 31;i ++)
    {
        if(ba[i+1] > key)
        {
            t1 = 1,t2 = 0;
            for(j = i-1;j >= 0;j --)//特殊讨论
            {
                if(ba[j]&key)
                {
                    t2 ++;
                    for(k = 0;k <= i+1-t1-t2;k ++)
                    {
                        if(k + t2 >= (i+1-k-t2))
                        ans += c[i+1-t1-t2][k];
                    }
                    t2 --;
                    t1 ++;
                }
                else
                t2 ++;
            }
            if(t2 >= t1)
            ans ++;
            break;
        }
        for(j = 0;j <= i;j ++)//第一位是1，还有i位。从i位中选j个0
        {
            if(2*j >= i+1)
            ans += c[i][j];
        }
    }
    return ans;
}
int main()
{
    int i,j;
    long long start,end;
    for(i = 0;i <= 31;i ++)
    c[i][0] = 1;
    for(i = 1;i <= 31;i ++)
    {
        for(j = 1;j <= 31;j ++)
        c[i][j] = c[i-1][j-1] + c[i-1][j];
    }
    ba[0] = 1;
    for(i = 1;i <= 31;i ++)
    ba[i] = 2*ba[i-1];
    scanf("%lld%lld",&start,&end);
    printf("%d\n",judge(end) - judge(start-1));
    return 0;
}