poj 3252 （转换成二进制的数位dp ）

题目链接：

http://poj.org/problem?id=3252

题目大意：

给你两个整数n，m，求n到m之间的数转换成二进制后，0大于等于1的数的个数。

解题思路：

状态设置：dp[pos][num0][num1]，pos表示当前位置，num0表示0的个数，num1表示1的个数。

目标状态：num0>=num1

注意事项，要多设置一个first来记录是否有前导0.

其余细节是数位dp的惯用套路，详见：

http://blog.csdn.net/qq_28302731/article/details/52173767

代码如下：

#include <iostream>
#include <string.h>

using namespace std;

int  dp[50][50][50];
int a[50];

//first记录是否有前导零。比如十进制数26是11010,若不加first判断，dfs比如搜到十进制数10，会出现01010，可是最前面的0显然是不能算的。
int dfs(int pos,int num0,int num1,int limit,int first)
{
    if(pos<1)
    {
        if(num0>=num1)
            return 1;
        else
            return 0;
    }
    if(!first&&!limit&&dp[pos][num0][num1]!=-1)
        return dp[pos][num0][num1];

    int End= limit? a[pos]:1 ;
    int ans=0;

    for(int i=0;i<=End;i++)
    {
        
        if(first)
        {
            if(i==0)
                ans += dfs(pos-1,0,0,limit&&(i==End),1);
            else
                ans += dfs(pos-1,num0,num1+1,limit&&(i==End),0);
        }
        else
        {
            if(i==0)
                ans += dfs(pos-1,num0+1,num1,limit&&(i==End),0);
            else
                ans += dfs(pos-1,num0,num1+1,limit&&(i==End),0);
        }

    }

    if(!limit&&!first)
        dp[pos][num0][num1]=ans;

    return ans;
}

int cal(int n)
{
    int len=0;
    while(n)
    {
        a[++len]=n&1;
        n=n>>1;
    }
    int ans=dfs(len,0,0,1,1);
    return ans;

}

int main()
{
    memset(dp,-1,sizeof(dp));
    int n,m;
    while(cin>>n>>m)
    {
        cout<<cal(m)-cal(n-1)<<endl;
    }
    return 0;
}