给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
思路:dfs。就按示例来说,n=4,k=2,即将1、2、3、4这4个数按照两个组合的方式,其组合方式有如上有条件。首先两个数字不能相同,即不能有(1,1)(2,2)这种的,其次还有顺序问题。
以1234为例,第一个元素为1开始,即有12,13,14,这个是没问题的,到了以2为第一个元素,则只有23,24,即没有21这个组合。
再到了以3为第一个元素,则只有了34,没有31和32。
所以,该组合的元素,只能是递增的(相等都不行),并且组合数为k。
DFS,遍历整个数组,情况1:如果当前为空(即添加当前组合的第一个元素) 或者 当前要添加的元素满足 大于 当前组合数组中的最后一个元素大小(因为该数组为升序),即可添加 。情况2:在情况1之前(即递归出口),如果当前组合的个数等于k,则添加到总的数组中去。
具体看代码 再加上上面的叙述,很好理解:
//#include<iostream>
//#include<vector>
//#include<algorithm>
//using namespace std;
class Solution {
public:
void dfs(vector<vector<int>> &re, int n, vector<int> &temp, int k) {
if (temp.size() == k) {
re.push_back(temp);
return;
}
for (int i = 0; i<n; ++i) {
if (!temp.empty() && i + 1 <= temp.back()) //如果以及添加了并且最后一个元素和当前要添加的不想等,比如1,1 2,2这种,直接continue;
continue;
temp.push_back(i + 1);//递归前修改
dfs(re, n, temp, k);
temp.pop_back();//递归后恢复
}
}
vector<vector<int>> combine(int n, int k) {
vector<int>temp;
vector<vector<int>> re;
dfs(re, n, temp, k);
return re;
}
};
/* 本地IDE自测
int main()
{
Solution a;
int n = 4, k = 2;
vector<vector<int>> re=a.combine(n, k);
return 0;
}
*/