组合总和III-回溯

最新推荐文章于 2024-10-13 15:10:00 发布

wjy_1126

最新推荐文章于 2024-10-13 15:10:00 发布

阅读量146

点赞数

分类专栏：算法 # 回溯文章标签：深度优先 leetcode 算法 java

本文链接：https://blog.csdn.net/qq_42692908/article/details/122898351

版权

算法同时被 2 个专栏收录

59 篇文章 0 订阅

订阅专栏

回溯

17 篇文章 0 订阅

订阅专栏

题目

找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数，并且每种组合中不存在重复的数字。

示例

输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]

代码

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    List<Integer> cur = new ArrayList<>();
    public void dfs(int k, int n, int index, int sum) {
        if (cur.size() == k) {
            if (sum == n) {
                res.add(new ArrayList<>(cur));
            }
            return;
        }
        for (int i = index; i <= 9; i++) {
            if (sum + i > n) {
                continue;
            }
            sum += i;
            cur.add(i);
            dfs(k,n,i+1,sum);
            sum -= i;
            cur.remove(cur.size() - 1);
        }
    }
    public List<List<Integer>> combinationSum3(int k, int n) {
        dfs(k,n,1,0);
        return res;
    }
}