1. 程式人生 > >LeetCode-47.全排列II(相關話題:回溯演算法)

LeetCode-47.全排列II(相關話題:回溯演算法)

給定一個可包含重複數字的序列,返回所有不重複的全排列。

示例:

輸入: [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);
        }
    }
}