poj 3252 Round Numbers

题目链接：http://poj.org/problem?id=3252

题意：给出一个二进制区间，求出0的个数不小于1的个数这样的二进制个数

解法：数位DP，定义状态dp[len][num_zero][num_one]，num_zero 定义为写0的个数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#define inf 0x7fffffff
#define exp 1e-10
#define PI 3.141592654
using namespace std;
typedef long long LL;
LL digit[35];
LL dp[35][35][35];
LL dfs(LL len,LL num_zero,LL num_one,bool fp,int flag)
{
    if (!len)
    {
        if (flag && num_zero>=num_one) return 1;
        return 0;
    }
    if (!fp && dp[len][num_zero][num_one]!=-1)
    return dp[len][num_zero][num_one];
    LL ret=0;
    LL fpmax= fp ? digit[len] : 1;
    for (LL i=0 ;i<=fpmax ;i++)
    {
        LL s= i==1 ? 1 : 0;
        LL s2= flag && i==0 ? 1 : 0;
        ret += dfs(len-1,num_zero+s2,num_one+s,fp && i==fpmax,flag || i==1);
    }
    if (!fp) return dp[len][num_zero][num_one]=ret;
    return ret;
}
LL f(LL n)
{
    LL len=0;
    while (n)
    {
        digit[++len]=n%2;
        n /= 2;
    }
    return dfs(len,0,0,true,0);
}
int main()
{
    LL a,b;
    while (cin>>a>>b)
    {
        memset(dp,-1,sizeof(dp));
        printf("%I64d
",f(b)-f(a-1));
    }
    return 0;
}