美文网首页
回溯算法之-子集

回溯算法之-子集

作者: 不知名的程序员 | 来源:发表于2021-03-14 18:45 被阅读0次

    关于回溯法的模版请看:https://www.jianshu.com/p/2a9856b96a86

    leetcode 78 子集

    给你一个整数数组 nums ,数组中的元素 互不相同 。返回该数组所有可能的子集(幂集)。解集 不能 包含重复的子集。你可以按 任意顺序 返回解集
    这个题和组合总和的题类似,只不过我们将解集收集的地方不同,套回溯法模版

    public List<List<Integer>> subsets(int[] nums) {
            List<List<Integer>> res = new ArrayList<>();
            Arrays.sort(nums);
            subsetsHelp(res, new ArrayList<>(), nums, 0);
            return res;
        }
    
        private void subsetsHelp(List<List<Integer>> res, ArrayList<Integer> list, int[] nums, int index) {
          // 在递归开始的地方将路径进行收集,一边回溯一边收集
            res.add(new ArrayList<>(list));
            for (int i = index; i < nums.length; i++) {
                list.add(nums[I]);
                subsetsHelp(res, list, nums, i + 1);
                list.remove(list.size()-1);
            }   
        }
    

    Leetcode 90 子集2

    给定一个可能包含重复元素的整数数组 nums,返回该数组所有可能的子集(幂集)。
    该题的解法和组合总和2基本相同

    public List<List<Integer>> subsetsWithDup(int[] nums) {
            List<List<Integer>> res = new ArrayList<>();
            Arrays.sort(nums);
            subsetsWithDupHelp(res, new ArrayList<>(), nums, 0);
            return res;
        }
    
        private void subsetsWithDupHelp(List<List<Integer>> res, ArrayList<Integer> list, int[] nums, int index) {
            res.add(new ArrayList<>(list));
            for (int i = index; i < nums.length; i++) {
                // 同一层若相同元素被使用过则跳过
                if (i > index && nums[i] == nums[i-1]) {
                    continue;
                }
                list.add(nums[I]);
                subsetsWithDupHelp(res, list, nums, i+1);
                list.remove(list.size()-1);
            }
        }
    

    相关文章

      网友评论

          本文标题:回溯算法之-子集

          本文链接:https://www.haomeiwen.com/subject/ffcccltx.html