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;
}