【leetcode刷题笔记】Subsets II

Given a collection of integers that might contain duplicates, S, return all possible subsets.

Note:

• Elements in a subset must be in non-descending order.
• The solution set must not contain duplicate subsets.

 

For example,
If S = [1,2,2], a solution is:

[
  [2],
  [1],
  [1,2,2],
  [2,2],
  [1,2],
  []
]

---

跟subsets这道题非常相像，除了S中会有重复的元素，那么会有什么影响呢？看一个例子：

S = [1,1]

根据subsets中的算法，生成的子集如下：

[
  [],
  [1],
  [1],
  [1,1]
]

显然[1],[1]这两个是重复的，需要去掉。那么我们怎么判重？只需要在递归函数的循环里面加上如下判断

 if(i != pos && s[i] == s[i - 1]) continue; 就可以了。

理解这个判断的关键，是搞清楚递归函数

public void subsetsDfs(int[] s,int level,int pos,ArrayList<Integer> result,List<List<Integer>> answer)

中pos变量的含义。

在这个函数中，level表示要生成的子集的长度，pos则表示目前要生成子集的第pos位上的元素。那么当i==pos的时候，说明子集中第pos位上的元素还没有生成过，那么即使s[i]==s[i-1]，它们在生成的子集中所占有的位置不同(例如例子中的[1,1]中的两个1分别在生成的子集的第0位和第1位上），s[i]就可以加到这个子集上去。但是当i != pos的时候，说明生成的子集上第pos位已经有元素了，如果当前的s[i]和这个元素s[i-1]相等，那么就没有必要再在这个位置上重复放置s[i]了(例如上述例子中已经有子集[1]的时候，不能够再生成子集[s[1]=1]了)。

最后实现的代码如下：

public class Solution {
    public void subsetsDfs(int[] s,int level,int pos,ArrayList<Integer> result,List<List<Integer>> answer){
        if(result.size() == level){
            answer.add(new ArrayList<Integer>(result));
            return;
        }
        for(int i = pos;i < s.length;i++){
            if(i != pos && s[i] == s[i - 1])
                continue;                
            result.add(s[i]);
            subsetsDfs(s, level, i+1, result, answer);
            result.remove(result.size()-1);
        }
    }
    public List<List<Integer>> subsetsWithDup(int[] S) {
        Arrays.sort(S);
        List<List<Integer>> answer = new ArrayList<List<Integer>>();
        ArrayList<Integer> result = new ArrayList<Integer>();
        
        for(int i = 0;i <= S.length;i++)
            subsetsDfs(S, i, 0, result, answer);
        return answer;
    }
}