Keshawn_lu's Blog

Leetcode 216. 组合总和 III

回溯
字数统计: 270阅读时长: 1 min
2020/09/11 Share

题目简介：

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

说明：

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

示例 1:

1
2
输入: k = 3, n = 7
输出: [[1,2,4]]

示例 2:

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

思路：

依然是回溯的做法，当元素数量 > k 或 和大于 n 时，直接return即可。

需要注意的是要从上个元素的下一个位置开始遍历(不能取重复的数字)。

代码如下：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
class Solution {
public:

    vector<vector<int>> res;

    void dfs(int k, int n, int pos, vector<int>& temp){

        if(temp.size() == k && n == 0){

            res.push_back(temp);
            return;
        }

        if(temp.size() > k || n < 0)
            return;

        for(int i = pos; i <= 9; i++){

            temp.push_back(i);

            dfs(k, n - i, i + 1, temp);

            temp.pop_back();
        }
    }

    vector<vector<int>> combinationSum3(int k, int n) {

        vector<int> temp;

        dfs(k, n, 1, temp);

        return res;
    }
};
CATALOG
  1. 1. 题目简介：
  2. 2. 思路：
  3. 3. 代码如下：
Total : 424
2023
2022
2021
2020
2019
动态规划 树 深度优先搜索 哈希表 数组 深度优先遍历 单调栈 字符串 双指针 栈 贪心 图 模拟 广度优先搜索 回溯 滑动窗口 二分查找 广度优先遍历 记忆化搜索 数学 链表 前缀和 双向链表 优先队列 拓扑排序 KMP 堆 字典树 递归 并查集 矩阵 位运算 双端队列 吴恩达团队NLP 吴恩达深度学习 线性表
杂谈 算法相关 PAT 2019推理小说阅读计划 NLP 深度学习 大三课程学习 大四课程学习 浙工商OJ 牛客网