HDU 4374 F(x) ( 数位dp )

链接 ：F(x)

题意：
F(x) 为 x 的个每一位对应的 2 的次数之和，求 0 —b 中 f(b) >f(a)的 数的数量。
思路：

1. 数位 dp 记录每一位的和，一开始用的 + 记录最后的结果会不会 大于 f(a)，但这样因为每组数据的 f(a)都不一样，dp要么得加一维记录 f(a)要么每次算完一组把 dp值初始化。但这样就和暴力没什么区别了，因为很难找到共有的状态，每次都必须重新找，然后就 t了。
2. 然后用减法可以很好的解决这个问题，从 f(a)的值开始，一位一位减，看到最后这个值是否大于 0 就行了，这样就可以有很多共有的状态了。这时 dp 记录的状态是，到第pos位f(a)剩余的值。

TLE代码：

#include<iostream>
#include<cstdio>
#include<map>
#include<math.h>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e4+7;
int dp[12][maxn],T,ca,A,b,x[15];
char a[15];
int dfs(int pos,int sum,int limit){
    if(pos<0){
       return sum<=A;
    }
    if(!limit&&dp[pos][sum]!=-1) return dp[pos][sum];
    int up= limit ? x[pos] : 9;
    int tmp=0;
    for(int i=0;i<=up;i++){
        tmp+=dfs(pos-1,sum+i*(1<<pos),limit&&i==x[pos]);
    }
    if(!limit) dp[pos][sum]=tmp;
    return tmp;
}
int solve(int b){
    int pos=0;
    while(b>0){
        x[pos++]=b%10;
        b/=10;
    }
    return dfs(pos-1,0,1);
}
int main (){
    cin>>T;
    while(T--){
         A=0;
         memset(dp,-1,sizeof(dp));
         scanf("%s",&a);
         cin>>b;
         int len=strlen(a);
         for(int i=0;i<len;i++){
             int v=a[len-i-1]-'0';
             A+=(1<<i)*v;
         }
         int ans=solve(b);
         cout<<"Case #"<<++ca<<": "<<ans<<endl;
    }
}

AC代码：

#include<iostream>
#include<cstdio>
#include<map>
#include<math.h>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e4+7;
int dp[12][2*maxn],T,ca,A,b,x[15];
char a[15];
int dfs(int pos,int sum,int limit){
    if(pos<0){
       return sum>=0;
    }
    if(sum<0) return 0;
    if(!limit&&dp[pos][sum]!=-1) return dp[pos][sum];
    int up= limit ? x[pos] : 9;
    int tmp=0;
    for(int i=0;i<=up;i++){
        tmp+=dfs(pos-1,sum-(i<<pos),limit&&i==x[pos]);
    }
    if(!limit) dp[pos][sum]=tmp;
    return tmp;
}
int solve(int b){
    int pos=0;
    while(b>0){
        x[pos++]=b%10;
        b/=10;
    }
    return dfs(pos-1,A,1);
}
int main (){
    cin>>T;
    memset(dp,-1,sizeof(dp));
    while(T--){
         A=0;
         scanf("%s",&a);
         cin>>b;
         int len=strlen(a);
         for(int i=0;i<len;i++){
             int v=a[len-i-1]-'0';
             A+=(v<<i);
         }
         int ans=solve(b);
         cout<<"Case #"<<++ca<<": "<<ans<<endl;
    }
}