地址https://leetcode-cn.com/problems/combinations/

給定兩個整數 n 和 k，返回 1 ... n 中所有可能的 k 個數的組合。

示例:

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

 

解答

觀察樣例發現
當在某一個位置選中一個數後 後面填充的數字必須比它大
這樣可以避免選中 1 5 組合後 又出現了 5 1 組合

使用DFS 完成上述操作
那麼 每個數字都有兩種可能 填充或者 略過
如果操作的值超過n 那麼就不再繼續
如果當前已經填充K個數字了 那麼放入答案不再繼續

class Solution {
public:
vector<vector<int>> ans;

void dfs(int n, int k, int idx, vector<int> v)
{
    if(v.size()>k) return;
    //剩下的元素 全選也不足以達到選擇K個元素 剪枝
    if( (v.size() + n-idx+1) < k  ) return;


    if (v.size() == k) {ans.push_back(v);return;}


    v.push_back(idx);
    dfs(n, k, idx 
 
+ 1, v);
    v.pop_back();

    dfs(n, k, idx + 1, v);
}

vector<vector<int>> combine(int n, int k) {
    vector<int> v;
    dfs(n, k, 1, v);
    return  ans;
}

};