POJ 3252 Round Numbers <数位dp（记忆化所搜）/组合数学>

题目：传送门

题目大意：输入两个数start和finish，求闭区间[start，finish]中Round Numbers（二进制表示中0的个数大于等于1的个数）的个数。

分析：有两种做法。
一种是数位dp，limit的作用是，如果当前pos位之前的更高位已经达到最大值（默认受bit[]限制），那么当前位就会受bit[]限制，否则，则当前位为0，1都可以，这是为了保证搜索的数不会比x大。lead表示第一位是否已经有一个1。用dp[][][]（dp[pos][c0][c1]：搜索到pos位且更高位包含c0个0、c1个1时的Round Numbers（以下采用缩写RN）数）存储已经求得的值，使用记忆化搜索避免子问题重复计算。

第二种做法是将此问题看做一个组合数学问题。若要求<=x的RN数，比如数字x的二进制表示长度为len，那么我们先求二进制表示长度小于len的RN数，这个比较容易，设$L \lt len$，那么最高位肯定放1，剩下L-1位可以放0和1，那么满足0的个数大于等于1的个数的数$N=$

$$\begin{cases} {L-1 \choose (L-1)/2+1}+{L-1 \choose (L-1)/2+2}+...+{L-1 \choose L-1}=\frac{2^{L-1}-{L-1 \choose (L-1)/2}}{2}, & \text{if $L$ is even} \\ {L-1 \choose (L-1)/2}+{L-1 \choose (L-1)/2+1}+...+{L-1 \choose L-1}=\frac{2^{L-1}}{2}, & \text{if $L$ is odd} \end{cases}$$

然后再加上长度为$L$的RN数，求长度为$L$的RN数的方法为，从x的高位向低位扫描，若当前位$position$是1，那么将当前位$position$放0，然后根据高位0和1的个数计算剩下位满足条件的放置数，具体看代码吧。

代码：

/*数位dp*/
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string>
using namespace std;

int dp[33][33][33];
int bit[33];

int dfs(int pos,int c0,int c1,bool lead,bool limit){
    if(pos==-1){
        if(lead) return 1;
        if(c0>=c1) return 1;
        return 0;
    }
    if(!lead&&!limit&&dp[pos][c0][c1]!=-1) return dp[pos][c0][c1];
    int up=limit?bit[pos]:1;
    int ans=0;
    for(int i=0;i<=up;++i){
        if(lead){
            if(!i) ans+=dfs(pos-1,0,0,lead,limit&&i==up);
            else ans+=dfs(pos-1,c0,c1+1,false,limit&&i==up); 
        }
        else{
            if(!i) ans+=dfs(pos-1,c0+1,c1,lead,limit&&i==up);
            else ans+=dfs(pos-1,c0,c1+1,lead,limit&&i==up);
        }
    }
    if(!lead&&!limit) dp[pos][c0][c1]=ans;
    return ans;
}

int solve(int x){
    int len=0;
    while(x) bit[len++]=x&1,x>>=1;
    return dfs(len-1,0,0,true,true);
}

int main(){
    int s,f;
    while(cin>>s>>f){
        memset(dp,-1,sizeof dp);
        cout<<solve(f)-solve(s-1)<<endl;
    }
    return 0;
}

/*组合数学*/
#include <iostream>
#include <cstring>
#include <string>
using namespace std;

int s,f;
int c[33][33];
int bit[33];

void calC(){
    c[0][0]=c[1][0]=c[1][1]=1;
    for(int i=2;i<33;++i){
        c[i][0]=1;
        for(int j=1;j<i;++j){
            c[i][j]=c[i-1][j-1]+c[i-1][j];
        }
        c[i][i]=1;
    }
}

int solve(int x){
    if(x<=1) return 0;
    int ans=0;
    int len=0;
    int c0=0,c1=0;
    while(x){
        if(x&1) bit[len++]=1,c1++;
        else bit[len++]=0,c0++;
        x>>=1;
    }
    if(c0>=c1) ans++;
    for(int i=1;i<len;++i){
        if(i&1) ans+=((1<<(i-1))-c[i-1][(i-1)/2])/2;
        else ans+=(1<<(i-1))/2; 
    }
    c0=0; c1=1;
    for(int i=len-2;i>=0;--i){
        if(bit[i]==1){
            for(int j=i;j>=0&&c0+j+1>=c1+i-j;--j){
                ans+=c[i][j];
            }
            c1++;
        }
        else c0++;
    }
    return ans;
}

int main(){
    ios::sync_with_stdio(false);
    calC();
    while(cin>>s>>f){
        cout<<solve(f)-solve(s-1)<<endl;
    }
    return 0;
}