POJ3252 Round Numbers 题解 数位DP

题目大意：
求区间 \([x,y]\) 范围内有多少数的二进制表示中的‘0’的个数 \(\ge\) ‘1’的个数。

解题思路：
使用 数位DP 解决这个问题。

我们设状态 f[pos][num0][num1][all0] 表示在：

• 当前所在数位为 pos ；
• 当前选择的‘0’的个数为 num0；
• 当前选择的‘1’的个数为 num1；
• 到当前位位置是不是前面的数都是前导零（如果都是前导0则 all0==true，否则 all==false）。

下的方案数。

我们开函数 dfs(int pos, int num0, int num1, bool all0, bool limit) 来解决这个问题。
其中，

• pos、num0、num1、all0 所表示的含义和上述表述一致，
• limit 表示当前是否处于限制条件。

实现代码如下：

#include <cstdio>
#include <cstring>
int f[33][33][33][2], a[33];
void init() {
    memset(f, -1, sizeof(f));
}
int dfs(int pos, int num0, int num1, bool all0, bool limit) {
    if (pos < 0)    // 遍历完所有数位
        return num0 >= num1 ? 1 : 0;
    if (num0 + pos + 1 < num1)  // 0的个数用于达不到1的个数
        return 0;
    if (!limit && f[pos][num0][num1][all0] != -1) return f[pos][num0][num1][all0];
    int up = limit ? a[pos] : 1;
    int tmp = 0;
    for (int i = 0; i <= up; i ++) {
        tmp += dfs(pos-1, num0 + (!all0 && i==0), num1 + (i==1), all0 && i==0, limit && i==up);
    }
    if (!limit) f[pos][num0][num1][all0] = tmp;
    return tmp;
}
int get_num(int x) {
    int pos = 0;
    while (x) {
        a[pos++] = x % 2;
        x /= 2;
    }
    return dfs(pos-1, 0, 0, true, true);
}
int main() {
    init();
    int x, y;
    while (~scanf("%d%d", &x, &y)) {
        printf("%d\n", get_num(y) - get_num(x-1));
    }
    return 0;
}