216组合总和 III(深度优先搜索递归回溯)

1、题目描述

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

说明:

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

2、示例

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

3、题解

基本思想:递归回溯,深度优先搜索。因为递归回溯所以把vector<int> cur设置为全局变量,另外在dfs中加入n - i >= 0可以减少一些情况继续dfs调用内存资源。

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
    vector<vector<int>> res;
    vector<int> cur;
    vector<vector<int>> combinationSum3(int k, int n) {
        //基本思想:递归回溯,深度优先搜索
        dfs( k, n, 1);
        return res;
    }
    void dfs(int k, int n, int num)
    {
        if (k == 0)
        {
            if (n == 0)
                res.push_back(cur);
            return;
        }
        for (int i = num; i < 10 && n - i >= 0; i++)
        {
            cur.push_back(i);
            dfs(k - 1, n - i, i + 1);
            cur.pop_back();
        }
        return;
    }
};
int main()
{
    Solution solute;
    int k = 5, n = 27;
    vector<vector<int>> res = solute.combinationSum3(k, n);
    for (int i = 0; i < res.size(); i++)
    {
        for_each(res[i].begin(), res[i].end(), [](const auto v) {cout << v << " "; });
        cout << endl;
    }
    return 0;
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章