18 - 带重复元素的子集

2017.9.14

写了一个新的函数 peekUnique(int[] nums,int k)。就是从一个数组中挑选k个数，有几种选择方案。

这样当求数组的子集的时候，就是k从1 到nums.length的peekUnique之和。

从n个数中选取k个数的公式为   f（n，k） = f（n-1，k）+f（n-1，k-1）；

public class Solution {
    /*
     * @param nums: A set of numbers.
     * @return: A list of lists. All valid subsets.
     */
 	public static ArrayList<ArrayList<Integer>> peekUnique(int[] nums,int k){
		ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
		if(nums == null || k > nums.length){
			return res;
		}
		if(k == 1){
			HashSet<Integer> set = new HashSet<Integer>();
			for(int i = 0; i<nums.length;i++){
				ArrayList<Integer> tmp = new ArrayList<Integer>();
				if(!set.contains(nums[i])){
					set.add(nums[i]);
					tmp.add(nums[i]);
					res.add(tmp);
				}
			}
		}
		else if(k == nums.length){
			ArrayList<Integer> tmp = new ArrayList<Integer>();
			for(int i = 0; i<nums.length;i++){
				tmp.add(nums[i]);
			}
			res.add(tmp);
		}
		else{
			int []newArr = Arrays.copyOf(nums, nums.length-1);
			ArrayList<ArrayList<Integer>> tmp1 = peekUnique(newArr,k);
			Iterator ite1 = tmp1.iterator();
			HashSet<ArrayList<Integer>> set = new HashSet<ArrayList<Integer>>();
			while(ite1.hasNext()){
				ArrayList<Integer> tmp = (ArrayList<Integer>)ite1.next();
				if(!set.contains(tmp)){
					set.add(tmp);
					res.add(tmp);
				}
			}
			
			ArrayList<ArrayList<Integer>> tmp2 = peekUnique(newArr,k-1);
			Iterator ite2 = tmp2.iterator();
			while(ite2.hasNext()){
				ArrayList<Integer> tmp = (ArrayList<Integer>)ite2.next();
				tmp.add(nums[nums.length -1]);
				if(!set.contains(tmp)){
					set.add(tmp);
					res.add(tmp);
				}	
			}
		}
		return res;
		
	}
	 public static List<List<Integer>> subsetsWithDup(int[] nums) {
        // write your code here
		List<List<Integer>> res = new ArrayList<>();
		ArrayList<Integer> tmp = new ArrayList<Integer>();
		res.add(tmp);
		if(nums.length == 0){
			return res;
		}
		Arrays.sort(nums);
		for(int i = 1;i<=nums.length;i++){
			ArrayList<ArrayList<Integer>> resTmp = peekUnique(nums,i);
			res.addAll(resTmp);
		}
		return res;
    }
}