poj 3252 Round Numbers

来源：互联网发布：淘宝美工教程百度云盘编辑：程序博客网时间：2024/05/28 23:19

这算是个简单数位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;}