LeetCode——深度优先搜索应用(子集1 、2)

1、给定一组不含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
输入: nums = [1,2,3]
输出:
[
[3],
[1],
[2],
[1,2,3],
[1,3],
[2,3],
[1,2],
[]
]

private List<List<Integer>> ans = new ArrayList<>();

    // 记录是否选择该元素
    private boolean[] v = new boolean[100];
    // 此处的index代表的是数组的索引
    private void robot(int index, int[] nums) {
        if (index >= nums.length) {

            List<Integer> temp = new ArrayList<>();
            for (int i = 0; i < nums.length; i++) {
                if (v[i]) {
                    temp.add(nums[i]);
                }
            }
            ans.add(temp);
            return;
        }

        // 以下两种情况中 true代表选择该元素   false代表不选择该元素
        v[index] = true;
        robot(index + 1, nums);
        v[index] = false;
        robot(index + 1, nums);
    }

    public List<List<Integer>> subSet(int[] nums) {
        robot(0, nums);
        return ans;
    }

2、给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
说明:解集不能包含重复的子集。
输入: [1,2,2]
输出:
[
[2],
[1],
[1,2,2],
[2,2],
[1,2],
[]
]

    private List<List<Integer>> ans = new ArrayList<>();

    // 记录是否选择该元素
    private boolean[] v = new boolean[100];


    // 此处的index代表的是数组的索引
    private void robot(int index, int[] nums) {
        if (index >= nums.length) {

            List<Integer> temp = new ArrayList<>();
            for (int i = 0; i < nums.length; i++) {
                if (v[i]) {
                    temp.add(nums[i]);
                }
            }
            ans.add(temp);
            return;
        }
        // 这个判断是  为了排除重复的元素
        if (index > 0 && nums[index] == nums[index-1] && v[index - 1] == false) {
            v[index] = false;
            robot(index + 1, nums);
        } else {
            // 以下两种情况中 true代表选择该元素   false代表不选择该元素
            v[index] = true;
            robot(index + 1, nums);
            v[index] = false;
            robot(index + 1, nums);
        }
    }


    public List<List<Integer>> subSet(int[] nums) {
        // 先对元素进行排序
        for(int i = 0; i < nums.length; i++) {
            for(int j = i+1; j < nums.length; j++) {
                if (nums[i] > nums[j]) {
                    int temp = nums[i];
                    nums[i] = nums[j];
                    nums[j] = temp;
                }
            }
        }
        robot(0, nums);
        return ans;
    }

猜你喜欢

转载自blog.csdn.net/reuxfhc/article/details/80471716