POJ 3252 数位DP

题意

如果一个数二进制表示0的数量大于等于1的数量，则称这个数为Round数。问某个区间内Round数的数量。

题解

数位DP，状态迁移还是比较难想的。我选择了网上大多数人的解决思路，记忆化搜索解决。
dp[len][num1][num0] len代表当前位置，num1代表1的个数，num0代表0的个数。
至于DFS函数，dfs(int len,int num1,int num0,bool first,bool up) 除了DP数组需要的，额外增加了两个变量，first和up。first代表前面无1，up代表处于上界状态。这两个状态是特殊状态，需要特别处理。（需要特别处理的原因很简单，UP状态下不能随便选择，需要根据上界的值来进行状态选择。至于First状态，First前面的0都不能参与计算，因此也需要特殊处理）
有了上述两个特殊状态，因此在DFS的时候也需要分情况进行DFS。按照是否为First进行分情况DFS，同时在状态转移的时候判断一下是否处于上界就可以了。

代码

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define INF 0x3f3f3f3f
#define LL long long
#define MAXN 100010
#define EPS 1e-10
#define MOD 100000000

using namespace std;
int dig[35];
int dp[35][50][50];

int dfs(int len,int num1,int num0,bool first,bool up){
    if(len==0){
        if(num0>=num1)
            return 1;
        else
            return 0;
    }
    if(!up&&!first&&dp[len][num1][num0]!=-1){
        return dp[len][num1][num0];
    }
    int n=up?dig[len]:1;
    int ans=0;
    UP(i,0,n+1){
        if(first){
            if(i==0){
                ans+=dfs(len-1,num1,num0,true,up&&(n==i));
            }else{
                ans+=dfs(len-1,num1+1,num0,false,up&&(n==i));
            }
        }else{
            if(i==0){
                ans+=dfs(len-1,num1,num0+1,false,up&&(n==i));
            }else{
                ans+=dfs(len-1,num1+1,num0,false,up&&(n==i));
            }
        }
    }
    if(!up&&!first)
        dp[len][num1][num0]=ans;
    return ans;
}

int solve(int x){
    MEM(dp,-1);
    int len=0;
    W(x>0){
        dig[++len]=x%2;
        x/=2;
    }
    return dfs(len,0,0,true,true);
}

int main(){
    int s,f;
    W(~scanf("%d%d",&s,&f)){
        printf("%d\n",solve(f)-solve(s-1));
    }
}