【LeetCode-216】组合总和 III

问题

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

说明:

  • 所有数字都是正整数。
  • 解集不能包含重复的组合。

示例

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

解答

class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backTrace(k, n, 1);
        return res;
    }
private:
    vector<vector<int>> res;
    vector<int> ans;
    void backTrace(int k, int n, int num) {
        if (n == 0) {
            if (ans.size() == k) res.push_back(ans);
            return; // 和等于n但数量小于k则剪枝
        }
        for (int i = num; i <= 9; i++) {
            if (num > n || ans.size() >= k) return; // 和超过n或者数量超过k则剪枝
            ans.push_back(i);
            backTrace(k, n - i, i + 1);
            ans.pop_back();
        }
    }
};

重点思路

这道题可以看做一颗九叉树的路径求和问题,只不过有一堆附加条件,在合适的地方加上就行。

posted @ 2021-03-08 22:05  tmpUser  阅读(33)  评论(0编辑  收藏  举报