【POJ3252】Round Numbers(数位DP)

记录一个菜逼的成长。。

题目链接：http://poj.org/problem?id=3252
题目大意：
如果一个数的二进制表示中0的个数大于等于1的个数，那么这个数是符合条件的。
给你左端点和右端点，问在这区间内符合条件的数有多少。

dp[pos][num0][num1] := 第一维表示从高位往低位枚举到pos的位置，第二维表示二进制中0的个数，第三维表示二进制中1的个数，结果为满足条件的数。
这题可以说是数位dp的模板题。

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <algorithm>
#include <cstdlib>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <list>
#include <deque>
#include <cctype>
#include <bitset>
#include <cmath>
using namespace std;
#define ALL(v) (v).begin(),(v).end()
#define cl(a,b) memset(a,b,sizeof(a))
#define bp __builtin_popcount
#define pb push_back
#define mp make_pair
#define fin freopen("D://in.txt","r",stdin)
#define fout freopen("D://out.txt","w",stdout)
#define lson t<<1,l,mid
#define rson t<<1|1,mid+1,r
#define seglen (node[t].r-node[t].l+1)
#define pi 3.1415926
#define exp  2.718281828459
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int,int> PII;
typedef pair<LL,LL> PLL;
typedef vector<PII> VPII;
const int INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
template <typename T>
inline void read(T &x){
    T ans=0;
    char last=' ',ch=getchar();
    while(ch<'0' || ch>'9')last=ch,ch=getchar();
    while(ch>='0' && ch<='9')ans=ans*10+ch-'0',ch=getchar();
    if(last=='-')ans=-ans;
    x = ans;
}
inline bool DBread(double &num)
{
    char in;double Dec=0.1;
    bool IsN=false,IsD=false;
    in=getchar();
    if(in==EOF) return false;
    while(in!='-'&&in!='.'&&(in<'0'||in>'9'))
        in=getchar();
    if(in=='-'){IsN=true;num=0;}
    else if(in=='.'){IsD=true;num=0;}
    else num=in-'0';
    if(!IsD){
        while(in=getchar(),in>='0'&&in<='9'){
            num*=10;num+=in-'0';}
    }
    if(in!='.'){
        if(IsN) num=-num;
            return true;
    }else{
        while(in=getchar(),in>='0'&&in<='9'){
                num+=Dec*(in-'0');Dec*=0.1;
        }
    }
    if(IsN) num=-num;
    return true;
}
template <typename T>
inline void write(T a) {
    if(a < 0) { putchar('-'); a = -a; }
    if(a >= 10) write(a / 10);
    putchar(a % 10 + '0');
}
/******************head***********************/
int dp[50][50][50];
int bit[50];
int dfs(int pos,int num0,int num1,int limit,int first)
//limit表示前一位是否是上界，first表示第一位是否放0，为1表示第一位为0；
{
    if(pos == -1){
        return num0 >= num1;
    }
    //如果上一位不是上界并且已经搜索过，则返回
    if(!limit && dp[pos][num0][num1] != -1)return dp[pos][num0][num1];
    //如果上一位是上界，这一位也应该是上界，否则就放1；
    int ed = (limit ? bit[pos] : 1);
    int ans = 0;
    for( int i = 0; i <= ed; i++ ){
        //第一位不放1并且这一位也不放1
        if(first && !i)ans += dfs(pos-1,0,0,limit && i == ed,1);
        else {
            //这一位放0；
            if(!i)ans += dfs(pos-1,num0+1,num1,limit && i == ed,0);
            //这一位放1；
            else ans += dfs(pos-1,num0,num1+1,limit && i == ed,0);
            //ans += dfs(pos-1,num0 + (i == 0),num1 + (i == 1),limit && i == ed,0);
        }
    }
    //如果上一位不是上界，保存结果；
    if(!limit)dp[pos][num0][num1] = ans;
    return ans;
}
int solve(int x)
{
    int len = 0;
    while(x){
        bit[len++] = x % 2;
        x /= 2;
    }
    return dfs(len-1,0,0,1,1);
}
int main()
{
    cl(dp,-1);
    int l,r;
    while(~scanf("%d%d",&l,&r)){
        printf("%d\n",solve(r) - solve(l-1));
    }
    return 0;
}