90. 子集 II

124 阅读1分钟

题目介绍

力扣90题:leetcode-cn.com/problems/su…

image.png

回溯

该题目就是[78. 子集]的变形,无非就是给定的数组nums中包含有相同元素,所以我们还是基于上一题,接下来就是进行去重工作了。我们可以使用一个数组visited来标识当前下标的元素是否已经使用过,接着利用树结构如下,本质就是前序遍历:

image.png

代码如下:

class Solution {
    private List<List<Integer>> ans;
    private List<Integer> path;

    public List<List<Integer>> subsetsWithDup(int[] nums) {
        ans = new ArrayList<>();
        path = new ArrayList<>();
        //首先排序,让相同的两个元素排到一起去,便于去重
        Arrays.sort(nums);
        int n = nums.length;
        // 使用 visited 数组来记录哪一个元素在当前路径中被使用了
        boolean[] visited = new boolean[n];
        // 开始回溯
        backtrace(nums, 0, visited, n);
        return ans;
    }

    private void backtrace(int[] nums, int start, boolean[] visited, int n) {
        // 首先加入当前路径
        ans.add(new ArrayList<>(path));
        // 从 start 开始遍历每一个元素,尝试加入路径中
        for (int i = start; i < n; ++i) {
            // 如果当前元素和前一个元素相同,而且前一个元素没有被访问,说明前一个相同的元素在当前层已经被用过了
            if (i > 0 && nums[i - 1] == nums[i] && !visited[i - 1]) {
                continue;
            }
            // 记录下来,用过了当前的元素
            visited[i] = true;
            path.add(nums[i]); // 放到路径中
            backtrace(nums, i + 1, visited, n); // 向下一个递归
            visited[i] = false; // 回溯
            path.remove(path.size() - 1);
        }
    }
}