题目链接:https://cn.vjudge.net/problem/POJ-3252
题目大意:求l到r之间(包括l和r),二进制表示0的个数大于等于1的个数的数有多少个。
分析:将数字转换成二进制,digit数组存放某个数字的二进制位,然后套用数位dp模板就可以了,dp[i][j][k]表示还剩多少位,0的个数位j,1的个数位k。
#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
ll dp[40][40][40];
int digit[40];
ll dfs(int pos, int num0, int num1, int lead, int limit)
{
if(pos < 0) return num0 >= num1;
if(!limit && !lead && dp[pos][num0][num1] != -1) return dp[pos][num0][num1];
int up = limit ? digit[pos] : 1;
ll ans = 0;
for(int i = 0; i <= up; i++)
{
if(lead)
{
if(i == 1) ans += dfs(pos - 1, num0, num1 + 1, 0, limit && i == up);
else ans += dfs(pos - 1, num0, num1, 1, limit && i == up);
}
else
{
if(i == 1) ans += dfs(pos - 1, num0, num1 + 1, 0, limit && i == up);
else ans += dfs(pos - 1, num0 + 1, num1, 0, limit && i == up);
}
}
if(!limit && !lead) dp[pos][num0][num1] = ans;
return ans;
}
ll solve(ll n)
{
int k = 0;
while(n)
{
digit[k++] = n & 1;
n >>= 1;
}
return dfs(k - 1, 0, 0, 1, 1);
}
int main()
{
ll l, r;
memset(dp, -1, sizeof dp);
while(~scanf("%lld %lld",&l, &r))
{
printf("%lld\n",solve(r) - solve(l - 1));
}
return 0;
}