HDU 4734 F(x) (数位DP+优化技巧)*

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=4734

题目大意

给定函数f的定义，问在f函数的定义下，
在0到B中有多少个数满足，小于等于f(A)的
关系并且A事先给定。

题目分析 

数位DP处理，但是注意到这题时限非常小，
朴素想法是dp[pos][sum],这样会超时，
然后考虑加一维度存储特定的f（A）但值会很大数组开不下，
看了题解的提示才明白了转换优化的操作。
即我把搜索的初始值搞成跟f（A）有关这样记忆化就可以针对全局了，
那么想到我们可以做减法，从f（A）开始，一旦小于零中途直接判零
且无法记忆化，其他的就是正常操作了，
感觉这个做法还是蛮有思想的~

#include<bits/stdc++.h>
using namespace std;

#define debug puts("YES");
#define rep(x,y,z) for(int (x)=(y);(x)<(z);(x)++)
#define ll long long

#define lrt int l,int r,int rt
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define root l,r,rt
#define mst(a,b) memset((a),(b),sizeof(a))
#define pii pair<int,int>
#define fi first
#define se second
#define mk(x,y) make_pair(x,y)
const int mod=1e9+7;
const int maxn=9e3+100;
const int ub=1e6;
ll powmod(ll x,ll y){ll t; for(t=1;y;y>>=1,x=x*x%mod) if(y&1) t=t*x%mod; return t;}
ll gcd(ll x,ll y){return y?gcd(y,x%y):x;}
/*
题目大意：
给定函数f的定义，问在f函数的定义下，
在0到B中有多少个数满足，小于等于f(A)的
关系并且A事先给定。

题目分析：
数位DP处理，但是注意到这题时限非常小，
朴素想法是dp[pos][sum],这样会超时，
然后考虑加一维度存储特定的f（A）但值会很大数组开不下，
看了题解的提示才明白了转换优化的操作。
即我把搜索的初始值搞成跟f（A）有关这样记忆化就可以针对全局了，
那么想到我们可以做减法，从f（A）开始，一旦小于零中途直接判零
且无法记忆化，其他的就是正常操作了，
感觉这个做法还是蛮有思想的~
*/
int x,y,v,dig[10],cnt=0;
int dp[10][maxn];
int f(int x){
    int ret=0,tmp=1;
    while(x) ret+=tmp*(x%10),tmp<<=1,x/=10;
    return ret;
}
int dfs(int pos,int fg,int zero,int last,int sum){
    if(pos<0) return sum>=0;
    if(sum<0) return 0;
    if(!fg&&dp[pos][sum]!=-1) return dp[pos][sum];
    int ret=0,s,ub=fg?dig[pos]:9;
    for(int i=0;i<=ub;i++)
        ret+=dfs(pos-1,fg&&i==ub,zero&&i==0,last-1,sum-(1<<last)*i);
    if(!fg) dp[pos][sum]=ret;
    return ret;
}
int solve(int x){
    cnt=0;while(x) dig[cnt++]=x%10,x/=10;
    return dfs(cnt-1,1,1,cnt-1,v);
}
int main(){
    mst(dp,-1);
    int t;scanf("%d",&t);
    for(int ca=1;ca<=t;ca++){
        scanf("%d%d",&x,&y);v=f(x);
        printf("Case #%d: %d\n",ca,solve(y));
    }
    return 0;
}