2022.01.19_每日一题 leetcode.78

题目描述 :

78. 子集

给你一个整数数组 nums ，数组中的元素 互不相同 。返回该数组所有可能的子集（幂集）。

解集 不能 包含重复的子集。你可以按 任意顺序 返回解集。

示例 1：

输入：nums = [1,2,3]
输出：[[],[1],[2],[1,2],[3],[1,3],[2,3],[1,2,3]]

示例 2：

输入：nums = [0]
输出：[[],[0]]

提示：

1 <= nums.length <= 10
-10 <= nums[i] <= 10
nums 中的所有元素 互不相同

思路 :

类似于leetcode.46. 全排列，可以使用回溯法去写

代码 :

class Solution {
    public List<List<Integer>> subsets(int[] nums) {
        int len = nums.length;
        int index = 0;
        List<List<Integer>> res = new ArrayList<>();
        Stack<Integer> stack = new Stack<>();
        res.add(new ArrayList<>(stack));
        f(res, stack, len, nums, index);
        return res;
    }

    public void f(List<List<Integer>> res, Stack<Integer> stack, int len, int[] nums, int index) {
        if (index == len) {
            return;
        }
        for (int i = index; i < len; i++, index++) {
            stack.push(nums[i]);
            res.add(new ArrayList<>(stack));
            f(res, stack, len, nums, index + 1);
            stack.pop();
        }
    }
}

运行截图 :

leetcode.46

题目描述 :

46. 全排列

给定一个不含重复数字的数组 nums ，返回其 所有可能的全排列 。你可以 按任意顺序 返回答案。

示例 1：

输入：nums = [1,2,3]
输出：[[1,2,3],[1,3,2],[2,1,3],[2,3,1],[3,1,2],[3,2,1]]

示例 2：

输入：nums = [0,1]
输出：[[0,1],[1,0]]

示例 3：

输入：nums = [1]
输出：[[1]]

提示：

1 <= nums.length <= 6
-10 <= nums[i] <= 10
nums 中的所有整数 互不相同

代码 ：

class Solution {
    public List<List<Integer>> permute(int[] nums) {
        Stack<Integer> stack = new Stack<>();
        List<List<Integer>> list = new ArrayList<>();
        int len = nums.length;
        boolean[] isUsed = new boolean[len];
        def(list, stack, isUsed, len, nums, 0);
        return list;
    }

    public static void def(List<List<Integer>> list, Stack<Integer> stack, boolean[] isUsed, int len, int[] nums, int index) {
        if (index == len) {
            list.add(new ArrayList(stack));
            return;
        }
        for (int i = 0; i < len; i++) {
            if (isUsed[i]) {
                continue;
            }
            isUsed[i] = true;
            stack.push(nums[i]);
            def(list, stack, isUsed, len, nums, index + 1);
            stack.pop();
            isUsed[i] = false;
        }
    }
}