LightOJ - 1032 Fast Bit Calculations (数位DP)

类型：

数位DP

题意：

[0,N]范围内所有数的二进制表示，问出现"11"的次数。（“111”计为两次） (0 ≤ N < 2³¹).

思路：

状态dp[i][d] 表示所有d开头的i位数（二进制）中出现11的次数

dp[i][d] = dp[i-1][0~1(end)] + (d==1 && j==1)*(2^i-2(nowx%2^i-2+1))

出口：dp[i][~] = 0

没啥难点。状态转移卡了一下。一定要清晰的描述出转移诶。

代码：

#include <cstdio>
#include <cstring>

int num[50];
long long dp[50][2];
long long nowx;

long long dfs(int i, int d, bool isQuery) {
    if (i == 1) {
        return 0;
    }
    long long &nowdp = dp[i][d];
    if (!isQuery && ~nowdp) return nowdp;
    int end = isQuery?num[i-1]:1;
    long long ans = 0;
    for (int j = 0; j <= end; j++) {
        ans += dfs(i-1, j, isQuery && j==end);
        if (d == 1 && j == 1) {
            ans += (isQuery && j==end)?(nowx%(1<<(i-2))+1):(1<<(i-2));
        }
    }
    if (!isQuery) nowdp = ans;
    return ans;
}

long long cal(long long x) {
    nowx = x;
    if (x == 0) return 0;
    int len = 0;
    while (x) {
        num[++len] = x%2;
        x>>=1;
    }
    return dfs(len+1, 0, true);
}

int main() {
    int t;
    scanf("%d", &t);
    int cas = 1;
    memset(dp, -1, sizeof(dp));
    while (t--) {
        long long n;
        scanf("%lld", &n);
        printf("Case %d: %lld\n", cas++, cal(n));
    }
    return 0;
}