题目
地址:https://leetcode.com/problems/permutations-ii/
Given a collection of numbers that might contain duplicates, return all possible unique permutations.
Example:
Input: [1,1,2]
Output:
[
[1,1,2],
[1,2,1],
[2,1,1]
]
回溯DFS实现
思路:
- 给数组nums排序;
- 用Boolean类型数组used记录每个位置是否已经用过。
- 重点条件
if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1])
,如果不是第一个位置,并且跟上一个位置的值一样,并且还没用过,说明不能再做头部位置。举个反例[1, 1, 2]
, 如果没有这个条件,则会出现两个结果。这个条件还把[2, 1, 1]
给去重,比如第二个1在位置1,因为条件!used[i - 1] == true
, 则成功去重。 - 正常情况是选择,
list.remove(list.size() - 1)
语句表示不选择,used[i] = false
并重置状态。
package backtracking;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
// https://leetcode.com/problems/permutations-ii/
public class PermutationsII {
public static void main(String[] args) {
PermutationsII obj = new PermutationsII();
int[] input = {1,1,2};
List<List<Integer>> resultList = obj.permuteUnique(input);
System.out.println(Arrays.toString(resultList.toArray()));
}
public List<List<Integer>> permuteUnique(int[] nums) {
List<List<Integer>> resultList = new ArrayList<List<Integer>>();
Arrays.sort(nums);
boolean[] used = new boolean[nums.length];
if (nums == null || nums.length == 0) {
return resultList;
}
dfs(nums, resultList, new ArrayList<Integer>(), used);
return resultList;
}
private void dfs(int[] nums, List<List<Integer>> resultList, List<Integer> list, boolean[] used) {
if (list.size() == nums.length) {
resultList.add(new ArrayList<Integer>(list));
return;
}
for (int i = 0; i < nums.length; i++) {
if (used[i]) {
continue;
}
if (i > 0 && nums[i] == nums[i - 1] && !used[i - 1]) {
continue;
}
used[i] = true;
list.add(nums[i]);
dfs(nums, resultList, list, used);
used[i] = false;
list.remove(list.size() - 1);
}
}
}