1. 程式人生 > >hdu 4374 F(x) 數位dp

hdu 4374 F(x) 數位dp

題意把一個int數二進位制拆分

然後給你一個a叫你輸出f(x)<=f(a)的所有值

我們咋做呢 看了別人的題解 用一個二維dp dp[pos][sum]代表到pos位置湊出sum需要多少值

我們要注意 sum一定是<=all的 所以如果sum>all 那麼直接return 0

然後由於不受前導0影響 所以直接用一個limit就行了

由於本質不會被查詢改變 所以我們在多組輸入外面用一次memset就可以了

/*
20181013
hdu 4734
*/
#include <iostream>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;
typedef long long ll;
int a[20];
ll dp[20][10005];
int all;
int f(int x){
    int ans = 0,pos = 0;
    while(x){
        ans += ((x%10)<<pos);
        pos++;
        x/=10;
    }
    return ans;
}
ll dfs(int pos,int sum,bool limit){
    if(pos==-1) return sum<=all;
    if(sum>all) return 0;
    if(!limit&&dp[pos][all-sum]!=-1) return dp[pos][all-sum];
    ll tmp = 0;
    int up = limit?a[pos]:9;
    for(int i = 0;i<=up;++i){
        tmp+=dfs(pos-1,sum+i*(1<<pos),limit&&i==a[pos]);
    }
    if(!limit) dp[pos][all-sum] = tmp;
    return tmp;
}
ll solve(int x){
    int pos = 0;
    while(x){
        a[pos++] = x%10;
        x/=10;
    }
    return dfs(pos-1,0,true);
}

int main(){
    int t;
    scanf("%d",&t);
    memset(dp,-1,sizeof(dp));
    for(int Case = 1;Case<=t;++Case){
        int a,b;
        scanf("%d%d",&a,&b);
        all = f(a);
        printf("Case #%d: %lld\n",Case,solve(b));
    }
    return 0;
}