POJ 3252 Round Numbers(数位DP)

题意:给一个区间[L,R],问在这个区间里有多少个数转化成二进制后0的个数>=1的个数
思路:
跟其他的数位DP有一点点不同
dp[i][j] 表示的是i位里有j个1有多少种情况
下面的数都是二进制的
以10110为例子
先统计0~1111的个数
然后从高位到低位统计
第一次遇到1时,就是把数看作10100,统计10000~10011的个数
然后第二次遇到1时,把数看作10110,统计10100~10101的个数
然后再加判断自己

#include<map>
#include<set>
#include<queue>
#include<stack>
#include<math.h>
#include<string>
#include<vector>
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm> 
using namespace std;
typedef long long LL;
#define maxn 1005
#define f(x) (x*1.0)
#define inf 0x3f3f3f3f
#define maxm maxn*maxn
#define min(a,b) (a>b?b:a)
#define max(a,b) (a>b?a:b)
#define lowbit(x) (x&(-x))
#define cheak(i) printf("%d ",i)
#define lson(x) (splay[x].son[0])
#define rson(x) (splay[x].son[1])
#define rfor(i,a,b) for(i=a;i<=b;++i)
#define lfor(i,a,b) for(i=a;i>=b;--i)
#define mem(a,b) memset(a,b,sizeof(a))
#define mec(a,b) memcpy(a,b,sizeof(b))
int dp[55][55],A[55];
void init()
{
    int i,j;
    rfor(i,0,50)
    {
        dp[i][0]=1;
        rfor(j,1,i)
        dp[i][j]=dp[i][j-1]*(i-j+1)/j;
    }
}
int solve(int x)
{
    if(x==-1) return 0;
    int len=0,ans=1,i,j,t=0,zero=0,one=1;
    while(x)
    {
        A[++len]=x&1;
        x/=2;
    }
    rfor(i,2,len-1)
    {
        //0~i/2-1
        for(j=0;j+1<=(i)/2;++j)
        {
            ans+=dp[i-1][j];
        }
    }
    //printf("***%d\n",ans);
    lfor(i,len-1,1)
    {
        if(A[i])
        {
            for(j=0;j<=i-1&j+one<=zero+(i-j);++j)
            ans+=dp[i-1][j];
            one++;
        }
        else zero++;
    }
    if(zero>=one) ans++;
    //printf("---%d\n",ans);
    return ans;
}
int main()
{
    int l,r;
    init();
    //printf("%d %d %d\n",dp[1][0],dp[2][0],dp[2][1]);
    while(~scanf("%d%d",&l,&r))
    {
        printf("%d\n",solve(r)-solve(l-1));
    }
    return 0;
} 

你可能感兴趣的:(poj,数位dp)