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

示例:

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

【分析】全排列,部分排列这些问题都是回溯的题目。只不过有些题目的解空间需要判断,有些的不需要。这道题的解空间是当长度等于k的大小时就是一个解,可以回溯到上层了。另一些题目如78. 子集 中,每个状态都是解。所以不需要做任何判断。

我们先来画图分析一下:

当n=5,k=3时的情况如下图:

leetcode 77 组合

从这个例子中,我们可以看到,在得到的组合的第一个位置,我们可以选择数字1-5。假设我们为组合的1个位置选择1,那么我们可以为第二个位置选择2-5。直到我们为所有位置选择好数字,我们就得到一个解。

当n=3,k=2时:

leetcode 77 组合

对于每个节点,往下递归时要分别对 n-level+1 颗子树都要做遍历,每遍历到叶子节点,回溯的时候记得pop掉当前尾部的元素,这样才能回到上一层的父节点。然后继续遍历其他子树。代码如下:

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> res;
        vector<int> out;
        helper(n,k,1,out,res);
        return res;
    }
    void helper(int n,int k,int level,vector<int>& out,vector<vector<int>>& res){
        if(out.size() == k){
            res.push_back(out);
            return;
        }
        for(int i = level; i <= n; ++i){
            out.push_back(i);
            helper(n,k,i+1,out,res);
            out.pop_back();
        }
    }
};

 图片参考:https://blog.csdn.net/u010500263/article/details/18435495

相关文章:

  • 2021-06-06
  • 2022-12-23
  • 2021-08-07
  • 2022-12-23
  • 2021-04-25
  • 2021-08-05
  • 2021-04-04
  • 2021-07-21
猜你喜欢
  • 2021-07-11
  • 2021-04-19
  • 2021-12-09
  • 2022-12-23
  • 2021-12-28
  • 2022-01-27
  • 2022-02-08
相关资源
相似解决方案