【数位DP】 HDU 4734 F(x)

原题直通车： HDU 4734 F(x)

题意：F(x) = A_n * 2^n-1 + A_n-1 * 2^n-2 + ... + A₂ * 2 + A₁ * 1, 求0.....B中F[x]<=F[A]的个数。

代码：

 

// 31MS	548K	931 B	G++

#include<iostream>

#include<cstdio>

#include<cstring>

using namespace std;

int digit[11], dp[11][6000], m;

inline int set_m(int x) {

    int k=1; m=0;

    while(x>0) m+=(x%10)*k, x/=10, k*=2;

}

inline int dfs(int pos, int rt, bool limit) {

    if(!pos) return (rt>=0);

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

    int len=(limit?digit[pos]:9);

    int ret=0;

    for(int i=0; i<=len; ++i) {

        int a=rt-i*(1<<(pos-1));

        if(a<0) break;

        ret+=dfs(pos-1, a, limit&&i==len);

    }

    if(!limit) dp[pos][rt]=ret;

    return ret;

}

inline int work(int x) {

    int len=0;

    while(x) digit[++len]=x%10, x/=10;

    return dfs(len, m, true);

}

int main() {

    int T, cas=1; scanf("%d",&T);

    memset(dp, -1, sizeof(dp));

    while(T--) {

        int x, r; scanf("%d%d",&x, &r);

        set_m(x);

        printf("Case #%d: %d\n", cas++, work(r));

    }

    return 0;

}