题目描述

给定两个整数 n 和 k，返回 1 ... n 中所有可能的 k 个数的组合。

示例:

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

算法1

DFS 剪枝可以加速
比如剩余的选择空间即使全选 也不足以满足条件 则剪枝

C++ 代码

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

void dfs(int n, int k, int idx, vector<int> v)
{
    //剩下的元素 全选也不足以达到选择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;
}

};