题目:Combinations
原题链接:https://leetcode.com/problems/combinations/
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],
]
给出两个整数 n 和 k ,返回所有可能的含有 k 个数字的组合,数字范围从 1 到 n 。
例如 n = 4, k = 2:
那么可能的组合有上图6种。
根据题意,数字的组合应该是不重复的。
整体思路应该从1开始每次选择一个数字,然后选择下一个数字,一直到选择的数字个数达到k,然后记录结果,一直到遍历结束,这是一个递归的过程。
注意,由于我们选择数字的思想是逐渐增加的,所以,可以在循环过程中做一些修改,舍弃肯定无用的循环,这样可以提高很多效率。
代码如下:
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> ans;
vector<int> temp;
insert(ans, temp, 1, n, k);
return ans;
}
void insert(vector<vector<int>>& ans, vector<int>& temp, int index, int n, int cnt) {
if(cnt == 0) {
ans.push_back(temp);
return;
}
// i 的上限不用 n ,而是选择n - cnt + 1,这是因为,如果当前第k - cnt个元素的选择如果大于
// n - cnt + 1的话,最多只剩下 cnt - 2个数字可以选择,而我们还需要选择cnt - 1个数字,这样
// 的话是肯定没有办法成立的,所以可以直接舍去。
for (int i = index; i <= n - cnt + 1; ++i) {
temp.push_back(i);
insert(ans, temp, i + 1, n, cnt - 1);
temp.pop_back();
}
}
};