【leetcode】【medium】216. Combination Sum III

最新推荐文章于 2024-08-22 10:11:02 发布

原创最新推荐文章于 2024-08-22 10:11:02 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

code 同时被 2 个专栏收录

233 篇文章

订阅专栏

c++

230 篇文章

订阅专栏

本文探讨了LeetCode上编号为216的问题“组合总和III”，该问题要求找出所有可能的k个数字组合，这些数字之和等于n，且仅限于1到9之间的数字。通过回溯法实现解决方案，文章详细解释了如何避免重复组合，并给出了清晰的代码示例。

216. Combination Sum III

Find all possible combinations of k numbers that add up to a number n, given that only numbers from 1 to 9 can be used and each combination should be a unique set of numbers.

Note:

All numbers will be positive integers.
The solution set must not contain duplicate combinations.

Example 1:

Input: k = 3, n = 7
Output: [[1,2,4]]

Example 2:

Input: k = 3, n = 9
Output: [[1,2,6], [1,3,5], [2,3,4]]

题目链接：https://leetcode-cn.com/problems/combination-sum-iii/

思路

法一：回溯法

和其他回溯法的题目做法类似，向下传递结果。

剪枝：因为是组合，且每个数只能用一次，则下一层迭代从上次取的数字之后开始。

终止条件是k==0（也可以是k==1）。

小坑：组合中的数字在[1,9]范围内。

class Solution {
public:
    vector<vector<int>> res;
    vector<vector<int>> combinationSum3(int k, int n) {
        if(k<=0 || n<=0 || n<k) return res;
        vector<int> cur;
        trace(k, n, 0, cur);
        return res;
    }
    void trace(int k, int n, int last, vector<int> &cur){
        if(k==0){
            if(n==0) res.push_back(cur);
            return;
        }
        for(int i = last+1; i<=9; ++i){
            if(i>n) break;
            cur.push_back(i);
            trace(k-1, n-i, i, cur);
            cur.pop_back();
        }
        return;
    }
};