POJ 3252 数位DP

链接：

http://poj.org/problem?id=3252

题意：

给你一个区间l，r，求区间中有多少个数转化为二进制后1的个数大于等于0的个数

题解：

还是数位dp，不过多了前导0的判断

代码：

int a[40];
int dp[40][80];

int dfs(int pos, int sta, bool lead, bool limit) {
    if (pos == -1) return sta >= 40;
    if (!lead && !limit && dp[pos][sta] != -1) return dp[pos][sta];
    int up = limit ? a[pos] : 1;
    int res = 0;
    rep(i, 0, up + 1) {
        if (lead && i == 0) res += dfs(pos - 1, sta, true, limit && i == a[pos]);
        else res += dfs(pos - 1, sta + (i == 0 ? 1 : -1), false, limit && i == a[pos]);
    }
    if (!lead && !limit) dp[pos][sta] = res;
    return res;
}

int solve(int x) {
    int pos = 0;
    while (x) {
        a[pos++] = x & 1;
        x >>= 1;
    }
    return dfs(pos - 1, 40, true, true);
}

int main() {
    int l, r;
    cin >> l >> r;
    memset(dp, -1, sizeof(dp));
    cout << solve(r) - solve(l - 1) << endl;
    return 0;
}