Leetcode: Permutations II
Given a collection of numbers that might contain duplicates, return all possible unique permutations. For example, [1,1,2] have the following unique permutations: [1,1,2], [1,2,1], and [2,1,1].
难度:80+15(怎样跳过重复的case来节省时间)。我一开始的想法很简单,就是在permutation这道题的基础上,在把合乎条件的permutation加入permutations的时候,加入一个查重语句。如果是重复的,就不添加。之前Subset II 问题我就是这么做的。
1 if (permutation.size() == num.length) { 2 if (!permutations.contains(permutation)) { 3 permutations.add(new ArrayList<Integer>(permutation)); 4 return; 5 } 6 }
可是这一次行不通了,报TLE。我就开始想,怎么节省时间呢?网上参看了别人的想法,说是对于重复的元素循环时跳过递归函数的调用。对题目这个例子来说就是,如果permutation已经计算过了[num[0], num[1], num[2]]即[1,1,2]这个case,那么之后如果permutation计算到[num[1], num[0], num[2]]又是[1,1,2]这个case的时候,需要跳过,不用算了。这样就可以剩下这部分递归的时间。
怎么知道当前的这个permutation之前出现过没有呢?(本题重点)
首先我们要对元素集合排序,从而让重复元素相邻,接下来就是一行代码对于重复元素和前面元素使用情况的判断。如果第一个重复元素前面的元素还没在当前结果中,那么我们不需要进行递归。
1 public class Solution { 2 public ArrayList<ArrayList<Integer>> permuteUnique(int[] num) { 3 Arrays.sort(num); 4 ArrayList<Integer> permutation = new ArrayList<Integer>(); 5 ArrayList<ArrayList<Integer>> permutations = new ArrayList<ArrayList<Integer>>(); 6 boolean[] visited = new boolean[num.length]; 7 helper(permutation, permutations, num, visited); 8 return permutations; 9 } 10 11 public void helper(ArrayList<Integer> permutation, ArrayList<ArrayList<Integer>> permutations, int[] num, boolean[] visited) { 12 if (permutation.size() == num.length) { 13 permutations.add(new ArrayList<Integer>(permutation)); 14 return; 15 } 16 17 for (int k = 0; k < num.length; k++) { 18 if (k > 0 && !visited[k-1] && num[k] == num[k-1]) continue; 19 if (!visited[k]) { 20 visited[k] = true; 21 permutation.add(num[k]); 22 helper(permutation, permutations, num, visited); 23 permutation.remove(permutation.size() - 1); 24 visited[k] = false; 25 } 26 } 27 } 28 }