hdoj 4734 数位dp（Java版）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4734
题目大意：题目给了个f(x)的定义：F(x) = An * 2n-1 + An-1 * 2n-2 + … + A2 * 2 + A1 * 1，Ai是十进制数位，然后给出a,b求区间[0,b]内满足f(i)<=f(a)的i的个数。
我的第二道数位dp题，相减的艺术
最开始数组开小了。。wa了
AC代码：

import java.util.Scanner;
/*
 * 常规想：这个f(x)计算就和数位计算是一样的，就是加了权值，所以dp[pos][sum]，这状态是基本的。a是题目给定的，f(a)是变化的不过f(a)最大好像是4600的样子。如果要memset优化就要加一维存f(a)的不同取值，那就是dp[10][4600][4600]，这显然不合法。
 *	这个时候就要用减法了，dp[pos][sum]，sum不是存当前枚举的数的前缀和(加权的)，而是枚举到当前pos位，后面还需要凑sum的权值和的个数，
 *	也就是说初始的是时候sum是f(a),枚举一位就减去这一位在计算f(i)的权值，那么最后枚举完所有位 sum>=0时就是满足的，后面的位数凑足sum位就可以了。
 *仔细想想这个状态是与f(a)无关的(新手似乎很难理解)，一个状态只有在sum>=0时才满足，如果我们按常规的思想求f(i)的话，那么最后sum>=f(a)才是满足的条件。
 *
 */
public class hdoj4734 {
	static int dp[][]=new int[12][10005];
	static int a[]=new int[12];
	static int all;
	static int f(int x){
		int res=0;
		int cnt=0;
		while(x!=0){
			int tmp=x%10;
			tmp<<=cnt;
			x/=10;
			cnt++;
			res+=tmp;
		}
		return res;
	}
	static int solve(int x){
		int pos=0;
		while(x!=0){
			a[++pos]=x%10;
			x/=10;
		}
		return dfs(pos,0,true);
	}
	static int dfs(int pos,int sum,boolean limit){
		if(pos==0){
			return sum<=all?1:0;
		}
		if(sum>all)
			return 0;
		if(!limit&&dp[pos][all-sum]!=-1)
			return dp[pos][all-sum];
		int up=(limit?a[pos]:9);
		int ans=0;
		for(int i=0;i<=up;i++){
			ans+=dfs(pos-1,sum+i*(1<<(pos-1)),limit&&a[pos]==i);
		}
		if(!limit)
			dp[pos][all-sum]=ans;
		return ans;
	}
	public static void main(String[] args) {
		Scanner sc=new Scanner(System.in);
		int t=sc.nextInt();
		for(int i=0;i<dp.length;i++){
			for(int j=0;j<dp[i].length;j++){
				dp[i][j]=-1;
			}
		}
		for(int i=1;i<=t;i++){
			int a,b;
			a=sc.nextInt();
			all=f(a);
			b=sc.nextInt();
			System.out.println("Case #"+i+": "+solve(b));
		}
		sc.close();
	}
}