LintCode 18.子集 II

描述

给定一个可能具有重复数字的列表，返回其所有可能的子集。

样例
样例 1：

输入：[0]
输出：
[
  [],
  [0]
]
样例 2：

输入：[1,2,2]
输出：
[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

挑战
你可以同时用递归与非递归的方式解决么？

注意事项

• 子集中的每个元素都是非降序的
• 两个子集间的顺序是无关紧要的
• 解集中不能包含重复子集

思路

在上一个的模板基础上加上限定，数组中有重复元素，首先排序，然后

if (i != 0 && nums[i] == nums[i - 1] && i > startIndex) {
    
    
	continue;
}

i ！= 0 防治越界，nums[i] == nums[i - 1] 判断正在要加入的元素是否和前一个元素相等， i > startIndex 判断前一个元素是否已经加入

题解

package LintCode;

public class Solution {
    
    
	/**
	 * @param nums: A set of numbers
	 * @return: A list of lists
	 */
	public static List<List<Integer>> subsets(int[] nums) {
    
    
		List<List<Integer>> results = new ArrayList<>();
		if (nums == null || nums.length == 0) {
    
    
			results.add(new ArrayList<>());
			return results;
		}

		Arrays.sort(nums);

		ArrayList<Integer> subset = new ArrayList<>();
		subsetsHelper(nums, 0, subset, results);

		return results;
	}

	private static void subsetsHelper(int[] nums, int startIndex, List<Integer> subset, List<List<Integer>> results) {
    
    

		results.add(new ArrayList<Integer>(subset));

		for (int i = startIndex; i < nums.length; i++) {
    
    
			if (i != 0 && nums[i] == nums[i - 1] && i > startIndex) {
    
    
				continue;
			}
			subset.add(nums[i]);
			subsetsHelper(nums, i + 1, subset, results);
			subset.remove(subset.size() - 1);
		}
	}
}