LeetCode 216. 组合总和 III（回溯法）

最新推荐文章于 2024-06-27 20:36:41 发布

givmchickenblood

最新推荐文章于 2024-06-27 20:36:41 发布

阅读量418

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/csdnlijingran/article/details/86236891

算法专栏收录该内容

69 篇文章

订阅专栏

本文深入探讨了组合总和III问题的解决方案，通过回溯算法寻找所有可能的k个数的组合，使它们的和等于给定的目标值n。文章详细介绍了如何使用递归回溯方法遍历所有可能的组合，同时避免重复和无效的计算，确保结果的正确性和效率。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目：216. 组合总和 III
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。
说明：
所有数字都是正整数。
解集不能包含重复的组合。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
思路

// 216. 组合总和 III
public class CombinationSum3 {
    public static void main(String[] args) {
        CombinationSum3 obj=new CombinationSum3();

        obj.combinationSum3(3,7);
    }
    public List<List<Integer>> combinationSum3(int k, int n) {
        List<List<Integer>> result = new ArrayList<>();
        int[] candidates = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9};
        Arrays.sort(candidates);
        backTrack(candidates, result, new ArrayList<Integer>(), n, k, 0);
        return result;
    }

    private void backTrack(int[] candidates, List<List<Integer>> result, ArrayList<Integer> list, int remain, int count, int start) {
        if (remain < 0) {
            return;
        } else if (remain == 0 && count == 0) {
            result.add(new ArrayList<>(list));
        } else if (count==0) {
            return;
        } else {
            for (int i = start; i < candidates.length; i++) {
                list.add(candidates[i]);
                //注意这里必须是count-1，而不能是count--或者--count，不然循环下一次方法的时候，count依然是3
                backTrack(candidates, result, list, remain - candidates[i], count-1, i + 1);
                list.remove(list.size() - 1);
            }
        }
    }
}