week_15_ Combinations

最新推荐文章于 2024-05-11 07:29:32 发布

原创最新推荐文章于 2024-05-11 07:29:32 发布 · 194 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

算法分析与设计专栏收录该内容

19 篇文章

订阅专栏

Description

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],
]

Solution

n个数选取k个排列组合，只需要进行k层递归即可，其思路如下：

① 从begin开始，对begin到n进行遍历，选取新元素i，加入组合result。

② 对result的当前大小与k进行比较，如果相等，则完成组合，将result加入最终的list。

③ 若不相等，则以新元素的下一元素为begin，进入下一层递归。

④ 将新元素从result中弹出，以对后面的元素继续进行遍历。

代码实现如下：

// 用递归进行解决
class Solution {
public:
	vector<vector<int>> combine(int n, int k) {
		vector<vector<int>> resultList;
		recursCombine(n, k, 1, resultList);
		return resultList;
	}

	void recursCombine(int n, int k, int begin, vector<vector<int>>& list) {
		for (int i = begin; i <= n; i++) {
			int new_element = i;
			result.push_back(new_element);
			if (result.size() == k)
				list.push_back(result);
			else
				recursCombine(n, k, i + 1, list);
			result.pop_back();
		}
	}
private:
	vector<int> result;
};

运行结果如下：