poj 3252 round number 【数位dp】

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

题意：求 [L , R]区间内 round number数的个数；
round numbe就是二进制0的个数大于等于1的个数的数字；

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<algorithm>
#include<math.h>
#include<map>
#include<set>
#include<queue>
#include<vector>
#include<stack>
#define inf 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int N=1e2+10;
const double pi=acos(-1.0);
const ll mod=1e9+7;

int dp[N][N][N];//第一维是位数，第二维记录0的个数，第三维记录一的个数
int a[N];

//pos表示位数，zero表示0的个数，one表示1的个数，limit表示是否到达数位上界
//重点是first，它表示二进制是否出现1，因为前面都是0的话并不是该数中二进制0的个数
int dfs(int pos,int zero,int one,bool limit,int first)
{
    if(pos==-1)
    {
        return zero>=one?1:0;
    }
    if(!limit&&dp[pos][zero][one]!=-1) return dp[pos][zero][one];
    int up=limit?a[pos]:1;
    int ans=0;
    for(int i=0;i<=up;i++)
    {
        int t=first;
        if(i==1) t=1;
        ans+=dfs(pos-1,t==1?zero+(i==0):0,t==1?one+(i==1):0,limit&&i==a[pos],t);
    }
    if(!limit) dp[pos][zero][one]=ans;
    return ans;
}

int solve(int x)
{
    int pos=0;
    while(x)
    {//十进制转换成二进制
        a[pos++]=x%2;
        x/=2;
    }
    return dfs(pos-1,0,0,true,0);
}

int main()
{
    int n,m;
    while(~scanf("%d %d",&n,&m))
    {
        memset(dp,-1,sizeof(dp));
        printf("%d\n",solve(m)-solve(n-1));
    }
    return 0;
}