https://vjudge.net/problem/POJ-3252
题意:给你2个数a,b,求[a,b]区间中的数满足2进制0>=1的个数. (1≤a<b≤ 2,000,000,000).
思路:用数位DP来解决,dp[i][count0][count1],表示搜索到第i位的时候,0与1的数量。记忆化搜索到底的时候如果0的数量大于等于1的数量,就返回1,表示在[a,b]区间存在这个数,结果+1。
要注意不能有前导0的情况。比如12化为二进制,其中一部分为000110,可以看出从高位往低位搜索时,会出现0比1多的情况,但二进制只有最高位为1时才会是一个数字。所以当出现前导为0得情况时,0的数量要重置为0;
#include<iostream>
#include<cstring>
using namespace std;
typedef long long ll;
ll dp[35][35][35];
int a[35];
ll dfs(int pos,int c0,int c1,bool limit,bool lead)
///lead==true, 前导为0;
{
if(pos==-1) return c0>=c1;
if(!limit&&dp[pos][c0][c1]!=-1) return dp[pos][c0][c1];
int up=limit?a[pos]:1;
ll ans=0;
for(int i=0;i<=up;i++)
{
ans+=dfs(pos-1,lead?0:c0+(i==0),c1+(i==1),limit&&i==up,lead&&!i);
}
if(!limit) dp[pos][c0][c1]=ans;
return ans;
}
ll slove(ll n)
{
int len=0;
while(n)
{
a[len++]=n%2;
n/=2;
}
return dfs(len-1,0,0,true,true);
}
int main()
{
ios::sync_with_stdio(false);
memset(dp,-1,sizeof(dp));
ll a,b;
cin>>a>>b;
cout<<slove(b)-slove(a-1)<<endl;
return 0;
}