简单的限定个数组合生成 Combinations

最新推荐文章于 2022-04-15 15:59:48 发布

原创最新推荐文章于 2022-04-15 15:59:48 发布 · 1.1k 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

151 篇文章

订阅专栏

我爱算法

142 篇文章

订阅专栏

本文介绍了一个使用回溯法解决组合问题的算法，用于生成指定范围内的所有组合，具体步骤包括定义组合类、实现组合生成函数以及演示实例。

问题：

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

For example,
If n = 4 and k = 2, a solution is:

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

思路：backtrack。由于状态容易now是引用类型传入传出的，所以用完之后要恢复原样。

class Solution {
public:
    vector<vector<int> > combine(int n, int k) {
        vector<vector<int> > com;
        if(n < k)
            return com;
        vector<int> now;
        produce(com, now, n, k, 0, 1);
        return com;
    }
    
    void produce(vector<vector<int> > &com, vector<int> &now, int n, int k, int count, int from)
    {
        if(count == k)
        {
            com.push_back(now);
            return;
        }
        if(from > n)
            return;
        
        produce(com, now, n, k, count, from+1);
        now.push_back(from);
        produce(com, now, n, k, count+1, from+1);
        now.pop_back(); //注意要将now改回原样
    }
};