leetcode刷题:回溯算法06(组合总和II)

64 阅读1分钟

本文已参与「新人创作礼」活动,一起开启掘金创作之路

40.组合总和II

力扣题目链接

给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用一次。

说明: 所有数字(包括目标数)都是正整数。 解集不能包含重复的组合。

示例 1: 输入: candidates = [10,1,2,7,6,1,5], target = 8, 所求解集为: [ [1, 7], [1, 2, 5], [2, 6], [1, 1, 6] ]

示例 2: 输入: candidates = [2,5,2,1,2], target = 5, 所求解集为: [ [1,2,2], [5] ]

获取结果的去重才麻烦哦。

 //剪枝操作,当 sum + candidates[i] >target 不进入
        for (int i = index; i < candidates.length; i++) {
            if (sum + candidates[i] > target) {
                break;
            }
            /**
             * 正确剔除重复解的办法
             * 跳过同一树层使用过的元素
             */
            if (i != index && candidates[i] == candidates[i - 1]) {
                continue;
            }
            list.add(candidates[i]);
            //    i+1 本元素不重复使用
            //    i   本元素可重复使用
            backtracking(candidates, sum + candidates[i], target, i + 1, list);
            list.remove(list.size() - 1);
        }

完整代码

package com.programmercarl.backtracking;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * @ClassName CombinationSum2
 * @Descriotion TODO
 * @Author nitaotao
 * @Date 2022/7/9 13:03
 * @Version 1.0
 * https://leetcode.cn/problems/combination-sum-ii/
 * 40. 组合总和 II
 **/
public class CombinationSum2 {
    List<List<Integer>> result = new ArrayList<List<Integer>>();

    public List<List<Integer>> combinationSum2(int[] candidates, int target) {
        //防止重复列表,保证生成列表唯一顺序
        Arrays.sort(candidates);
        backtracking(candidates, 0, target, 0, new ArrayList<>());
        return result;
    }

    /**
     * @param candidates 数字数组
     * @param sum        总和
     * @param target     目标值
     * @param index      递归索引起始值
     * @param list
     */
    public void backtracking(int[] candidates, int sum, int target, int index, List<Integer> list) {
        if (sum == target) {
            result.add(new ArrayList<>(list));
            return;
        }
        //剪枝操作,当 sum + candidates[i] >target 不进入
        for (int i = index; i < candidates.length; i++) {
            if (sum + candidates[i] > target) {
                break;
            }
            /**
             * 正确剔除重复解的办法
             * 跳过同一树层使用过的元素
             */
            if (i != index && candidates[i] == candidates[i - 1]) {
                continue;
            }
            list.add(candidates[i]);
            //    i+1 本元素不重复使用
            //    i   本元素可重复使用
            backtracking(candidates, sum + candidates[i], target, i + 1, list);
            list.remove(list.size() - 1);
        }
    }

    public static void main(String[] args) {
        System.out.println(new CombinationSum2().combinationSum2(new int[]{10, 1, 2, 7, 6, 1, 5}, 8));
    }
}

在这里插入图片描述

附上大佬链接 代码随想录