LeetCode-47.全排列II(相關話題:回溯演算法)
阿新 • • 發佈:2019-02-04
給定一個可包含重複數字的序列,返回所有不重複的全排列。
示例:
輸入: [1,1,2]
輸出:
[
[1,1,2],
[1,2,1],
[2,1,1]
]
解題思路:這個題跟上一道題類似,唯一的區別在於要排除重複元素,這裡利用了HashMap,存每層已搜尋過的元素,在每一層往下搜尋之前,查詢當前元素是否在當前搜尋層的HashMap中,在的話就跳過(這裡個人感覺有點像深度優先搜尋)。
java程式碼:
class Solution { public List<List<Integer>> permuteUnique(int[] nums) { List<List<Integer>> result = new LinkedList<>(); List<Integer> tmp = new ArrayList<>(nums.length); boolean[] flag = new boolean[nums.length]; dfs(nums, flag, tmp, 0, result); return result; } private void dfs(int[] nums, boolean[] flag, List<Integer> tmp, int cnt, List<List<Integer>> result) { if(cnt == nums.length) { result.add(tmp); return; } Map<Integer, Integer> f = new HashMap<>(); for(int i = 0; i < nums.length; i++) { if(flag[i] || null != f.get(nums[i])) continue; f.put(nums[i], i); boolean[] flag1 = Arrays.copyOf(flag, flag.length); List<Integer> tmp1 = new ArrayList<>(nums.length); tmp1.addAll(tmp); tmp1.add(nums[i]); flag1[i] = true; dfs(nums, flag1, tmp1, cnt+1, result); } } }