题意:给定范围[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;
}