poj 3252 Round Numbers

poj 3252 Round Numbers

这算是个简单数位dp题目

dp[pos][zeros][ones] 表示有pos 个自由位， 前面有zeros个零ones个1,的个数……要注意前导零的情况

#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

int dp[32][32][32],bit[32];

int dfs(int pos,int zeros,int ones,bool ok,bool doing)
{
    if(pos<0){
         //cout<<ok<<" "<<zeros<<" "<<ones<<endl;
         if(ok) return zeros>=ones;
         else return 0;
    }
    if(!doing&&dp[pos][zeros][ones]!=-1) return dp[pos][zeros][ones];

    int end=doing?bit[pos]:1;
    int ret=0;
    for(int i=0;i<=end;i++)
    {
        int nz=zeros,no=ones;
        if(ok||i) nz=nz+(i==0),no=no+i;
        ret+=dfs(pos-1,nz,no,ok||i,doing&&i==end);
    }
    if(!doing) dp[pos][zeros][ones]=ret;
    return  ret;
 }
int solve(int n)
{
    int pos=0;
    for(;n;n>>=1,pos++)
       bit[pos]=n&1;
    return dfs(pos-1,0,0,0,1);
}
int main()
{
    memset(dp,-1,sizeof(dp));
    int l,r;
    while(scanf("%d%d",&l,&r)==2)
      printf("%d\n",solve(r)-solve(l-1));
    return 0;
}