微软面试题:带重复元素的排列
描述
给出一个具有重复数字的列表,找出列表所有不同的排列。
在线评测地址
样例 1:
输入:
nums = [1,1]
输出:
[
[1,1]
]
样例 2:
输入:
nums = [1,2,2]
输出:
[
[1,2,2],
[2,1,2],
[2,2,1]
]
解题思路:
这道题我们需要使用dfs+回溯的方法来进行求解。
我们定义dfs函数,使用递归的方法对决策树进行深度优先遍历。对于长度为n的数组nums,我们一位一位地生成它的排列数组,每深入一层数组长度就加1,遍历到叶节点时生成数组的长度达到n,即为我们的答案。
由于数组中有重复元素,所以我们在遍历时需要剪枝操作。
源代码:
public class Solution {
/*
* @param : A list of integers
* @return: A list of unique permutations
*/
public List<List<Integer>> permuteUnique(int[] nums) {
List<List<Integer>> results = new ArrayList<>();
if (nums == null) {
return results;
}
Arrays.sort(nums);
dfs(nums, new boolean[nums.length], new ArrayList<Integer>(), results);
return results;
}
private void dfs(int[] nums,
boolean[] visited,
List<Integer> permutation,
List<List<Integer>> results) {
if (nums.length == permutation.size()) {
results.add(new ArrayList<Integer>(permutation));
return;
}
for (int i = 0; i < nums.length; i++) {
if (visited[i]) {
continue;
}
if (i > 0 && nums[i] == nums[i - 1] && !visited[i - 1]) {
continue;
}
permutation.add(nums[i]);
visited[i] = true;
dfs(nums, visited, permutation, results);
visited[i] = false;
permutation.remove(permutation.size() - 1);
}
}
};