HDU 4734 F(x) (数位DP)

F(x)

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
using namespace std;

int A, B;
//dp[pos][num]: 枚举到当前pos位，后面还需要凑num的权值和的个数
//也就是说初始的是时候num是f(a), 枚举一位就减去这一位在计算f(i)的权值，
//那么最后枚举完所有位 num>=0时就是满足的，后面的位数凑足num位就可以了。
int dp[20][200000];
int bit[20];

int dfs(int pos, int num, int limit) {
    if (pos == -1)
        return num >= 0;

    if (num < 0)//提前结束，不必再枚举了，优化
        return 0;

    if (!limit && dp[pos][num] != -1)
        return dp[pos][num];

    int ans = 0;
    int up = limit ? bit[pos] : 9;
    for (int i = 0; i <= up; i++)
        ans += dfs(pos-1, num-i*(1<<pos), limit && i == bit[pos]);

    if (!limit)
        dp[pos][num] = ans;

    return ans;
}

int F(int x) {
    int tmp = 0;
    int len = 0;
    while (x) {
        tmp += (x%10)*(1<<len);
        len++;
        x /= 10;
    }
    return tmp;
}

int cal() {
    int len = 0;
    while (B) {
        bit[len++] = B%10;
        B /= 10;
    }

    return dfs(len-1, F(A), 1);
}

int main() {

    freopen("data.in", "r", stdin);
    int t;
    int cas = 1;
    scanf("%d", &t);
    memset(dp, -1, sizeof(dp));

    while (t--) {
        scanf("%d%d", &A, &B);
        printf("Case #%d: %d\n", cas++, cal());
    }

    return 0;
}