Leetcode Combinations

最新推荐文章于 2025-12-15 23:25:39 发布

原创最新推荐文章于 2025-12-15 23:25:39 发布 · 226 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

124 篇文章

订阅专栏

本文介绍了一种使用递归方法生成所有可能的组合方案的算法，并通过剪枝优化减少不必要的计算分支，提高了算法效率。

Given two integers n and k, return all possible combinations ofk 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],
]

第一下想到的方法就是递归大法。。。

代码如下：

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> result;
        vector<int> temp;
        combine(result,temp,n,1,k);
        return result;
    }
    
    void combine(vector<vector<int>>& result,vector<int>& temp,int n,int start,int k)
    {
        if(temp.size() == k)
            result.push_back(temp);
        else
        {
            for(int i=start;i<=n;i++)
            {
                temp.push_back(i);
                combine(result,temp,n,i+1,k);
                temp.pop_back();
            }
        }
    }
};

发现还可以对代码进行一些改进，进行剪枝，通过对for循环的改进去除了一些不必要的分支。

代码如下：

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> result;
        vector<int> temp;
        combine(result,temp,n,1,k);
        return result;
    }
    
    void combine(vector<vector<int>>& result,vector<int>& temp,int n,int start,int k)
    {
        if(k == 0 | n == 0)
        {
            result.push_back(temp);
            return;
        }
        else
        {
            for(int i=start;i<=n-k+1;i++)
            {
                temp.push_back(i);
                combine(result,temp,n,i+1,k-1);
                temp.pop_back();
            }
        }
    }
};