LeetCode 47. 全排列 II

一、题目

二、解题

注：本文均是Java代码

1、回溯法（递归+剪枝）

注意要先排序

class Solution {
    public List<List<Integer>> permuteUnique(int[] nums) {
        Arrays.sort(nums);
        List<List<Integer>> result = new ArrayList<>();
        dfs(nums, new boolean[nums.length], new LinkedList<Integer>(), result);
        return result;
    }
    private void dfs(int[] nums, boolean[] visited, LinkedList<Integer> stack, List<List<Integer>> result) {
        if (stack.size() == nums.length) {
            result.add(new ArrayList<>(stack));
            return;
        }
        for (int i = 0; i < nums.length; i++) {
            // 找到与它想到的上一个数字，访问过进行递归，没有被访问过不进行递归
            if (i > 0 && nums[i] == nums[i-1] && !visited[i-1]) {
                continue;
            }
            if (!visited[i]) {
                stack.push(nums[i]);
                visited[i] = true;
                dfs(nums,visited,stack,result);
                stack.pop();
                visited[i] = false;
            }
        }
    }
}

2、回溯法（递归+剪枝）优化版

result、res、used定义为全局变量，循环中只有一个if。

class Solution {
    List<List<Integer>> result = new ArrayList<>();
    List<Integer> res = new ArrayList<>();  
    boolean[] used;
    public List<List<Integer>> permuteUnique(int[] nums) {
        Arrays.sort(nums);
        used = new boolean[nums.length];
        dfs(nums,0);
        return result;
    }
    private void dfs(int[] nums, int index) {
        if (index == nums.length) {
            result.add(new ArrayList<>(res));
        }
        for (int i = 0; i < nums.length; i++) {
            if (used[i] || (i > 0 && nums[i] == nums[i-1] && !used[i-1])) {
                continue;
            }
            res.add(nums[i]);
            used[i] = true;
            dfs(nums,index+1);
            res.remove(index);
            used[i] = false;
        }
    }
}