POJ - 3252 E - Round Numbers

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;
}