POJ 3252 Round Numbers

题意:给定范围[L,R],求在这个范围里换算成二进制以后0的个数比1多的数有多少个.
做法:数据范围2,000,000,000 O(n)也会超时,因为可以枚举数的每一位来解决问题,并且一个数是否满足题意与题目给的范围无关,答案满足前缀和形式:即可用solve®-solve(L-1),所以考虑用数位dp
dp[pos][num]表示pos位二进制数,0比1多num的有多少个.这里因为在计算过程中可能出现1比0多的情况,但只要到最后0比1多也符合要求,但是数组下标不能出现负数,所以我们将num的初始定为32,把32当作0(因为2,000,000,000大约是2的32次方,1最多比0多32个),如果最后num大于32说明0的个数大于等于1,否则就不满足要求.
这里要注意前导0的问题,有前导0时前导0不能算.

#include<iostream>
#include<cstring>
#include<cstdio>

using namespace std;

typedef long long ll;
ll dp[40][140]; 
ll a[40];
ll n,M;
ll dfs(ll pos,ll sta,bool limit,bool lead)
{
	if(pos==-1) {
		if (sta>=32)  return 1;
		return 0;
	}
	if(!limit && !lead&&dp[pos][sta]!=-1) return dp[pos][sta];
	//if(!limit && dp[pos][m/2-num+1]!=-1 && m%2==1) return dp[pos][m/2-num+1];
    ll tmp = 0;
    ll up = limit ? a[pos]:1;
	 
    for (ll i = 0 ;i <= up;i++)
    {
      if(!i && lead) tmp+=dfs(pos-1,sta,limit&&a[pos]==i,lead);
	  else   tmp+=dfs(pos-1,sta+(i==0?1:-1),limit&&a[pos]==i,lead && i==0);	
	}
	if(!limit && !lead) dp[pos][sta] = tmp;
	//if(!limit && m%2==1) dp[pos][m/2-num+1] = tmp;
	return tmp;
}
ll solve(ll x)
{   int pos = 0;
	while(x!=0)
	{
		if(x&1)
		{
			a[pos++] = 1;
			
		}
		else a[pos++] = 0;
      x>>=1;
	}
	dfs(pos-1,32,true,true);
}
int main(){
cin>>M>>n;
memset(dp,-1,sizeof(dp));
ll ans = solve(n)-solve(M-1);
cout<<ans;
return 0;
}