JAVA程序设计：最大为 N 的数字组合（LeetCode：902）

我们有一组排序的数字 D，它是  {'1','2','3','4','5','6','7','8','9'} 的非空子集。（请注意，'0' 不包括在内。）

现在，我们用这些数字进行组合写数字，想用多少次就用多少次。例如 D = {'1','3','5'}，我们可以写出像 '13', '551', '1351315' 这样的数字。

返回可以用 D 中的数字写出的小于或等于 N 的正整数的数目。

 

示例 1：

输入：D = ["1","3","5","7"], N = 100
输出：20
解释：
可写出的 20 个数字是：
1, 3, 5, 7, 11, 13, 15, 17, 31, 33, 35, 37, 51, 53, 55, 57, 71, 73, 75, 77.
示例 2：

输入：D = ["1","4","9"], N = 1000000000
输出：29523
解释：
我们可以写 3 个一位数字，9 个两位数字，27 个三位数字，
81 个四位数字，243 个五位数字，729 个六位数字，
2187 个七位数字，6561 个八位数字和 19683 个九位数字。
总共，可以使用D中的数字写出 29523 个整数。
 

提示：

D 是按排序顺序的数字 '1'-'9' 的子集。
1 <= N <= 10^9

思路：对于位数小于N的位数，可以直接预处理出来，而对于位数与N位数相同的情况，我们可以采用数位dp来搞。

不了解数位dp可以移位这里。

class Solution {
	
	int[] arr;
	int[] dp=new int[10];
	List<Integer> list=new ArrayList<>();
	
    public int atMostNGivenDigitSet(String[] D, int N) {
    	
    	for(int i=0;i<10;i++)
    			dp[i]=-1;
    	
    	while(N>0) {
    		list.add(N%10);
    		N/=10;
    	}
    	
    	int len=list.size();
    	arr=new int[D.length];
    	
    	for(int i=0;i<D.length;i++)
    		arr[i]=Integer.valueOf(D[i]);
    	
    	int ans=0;
    	for(int i=1;i<len;i++)
    		ans+=work(D.length,i);
    	ans+=DP(len,true);
    	
    	return ans;
    }
    
    private int DP(int pos,boolean limit) {
    	int ans=0;
    	if(pos==0) return 1;
    	if(!limit && dp[pos]>=0)
    		return dp[pos];
    	int num=limit?list.get(pos-1):9;
    	for(int i=0;i<arr.length && arr[i]<=num;i++)
    		ans+=DP(pos-1,limit && arr[i]==num);
    	if(!limit) dp[pos]=ans;
    	return ans;
    }
    
    private int work(int x,int y) {
    	int res=1;
    	for(int i=1;i<=y;i++)
    		res*=x;
    	return res;
    }
}