POJ - 3252 E - Round Numbers(数位dp)
题目链接:https://cn.vjudge.net/contest/163023#problem/E
题目大意:求l~r范围内的整数转化成二进制形式0和1个数相等的数的个数
题目分析:数位dp。挺好想的,关键是前导零不算是0,就是说0001100,1的个数是2,0的个数也是2。主要是要有一个变量标记一下。
#include <cstdio>
#include <iostream>
#include <algorithm>
using namespace std;
int data[30],dp[30][30][30];
int dfs(int pos, int num0, int num1, bool limit, bool first)
{
if(pos == -1 && num0 >= num1) return 1;
if(pos == -1 && num0 < num1) return 0;
if(!limit && !first && dp[pos][num0][num1]) return dp[pos][num0][num1];
int up = limit ? data[pos] : 1;
int ans = 0;
for(int i = 0; i <= up; i++)
{
if(first)
{
if(i == 0)
ans += dfs(pos - 1, 0, 0, i == data[pos] && limit, true);
else ans += dfs(pos - 1, 0/*num0*/, num1 + 1, i == data[pos] && limit, false);
}
else
{
if(i == 0)
ans += dfs(pos - 1, num0 + 1, num1, i == data[pos] && limit, false);
else ans += dfs(pos - 1, num0, num1 + 1, i == data[pos] && limit, false);
}
}
if(!limit && !first) dp[pos][num0][num1] = ans;
return ans;
}
int solve(int n)
{
int pos = 0;
while(n)
{
data[pos++] = n % 2;
n /= 2;
}
return dfs(pos - 1, 0, 0, true, true);
}
int main()
{
int l, r;
while(~scanf("%d %d", &l, &r))
{
printf("%d\n", solve(r) - solve(l - 1));
}
return 0;
}