【Leecode刷题】77题. Combinations
题目:
Given two integers n and k, return all possible combinations of k numbers out of 1 … n.
测试样例:
Input: n = 4, k = 2
Output:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
难度:middle
使用语言:c++
题目分析:
题目给定一个数n,以及一个数k,返回一个在[1,n]范围内包含k个数的组合。
我们就以最简单的思路去想,例如给定n=8,k=5,那么就排列就会是12345,12346,12347,12348,12356,12357,12358…
从这里可以很清楚的看到,每回我们在做的,只是让最后一位数字自增到在该位置的最大值,
(其实每个位置的最大值都是可以确定的,例如在这个例子当中,index=0时,数字在[1,4]之间,index=1时,数字在[2,5]之间,以此类推,对于任意的n,k,index,可以知道,该index上,数字范围是[index+1,n-k+index+1]),有点类似与计数时候的进位,所以,程序就好写了,每回都增加最后一位数字,满足条件就加入vector,不满足就增加前一位的数字,如果前一位也不能再增加了,就加前一位的前一位,在找到全部组合前,总会有一个位置上的数字可以增加,增加了位置的数字后,要把这个位置往后的数字初始化一下,例如12678,后三位都各自达到了最大值,那只能增加第二位的数字,也就是变成13678,然后初始化一下,变成13456(后一位比前一位的值大一)。问题得解。
完整代码:
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> res;
int helper[k];
for(int i = 1;i<=k;i++)
helper[i-1] = i; //初始化为[1...k]
int index = k-1; //最后一位的下标
while(index >= 0){ //首位数字还没达到最大值
if(helper[index] <= n-k+index+1){ //某个位置上的数字还没达到最大值
/*看某个位置的后一位数字是否达到最大值,如果是,说明要
*变更该位置,然后初始化
*/
if(index + 1 < k && helper[index+1] >= n-k+index+2){
if(helper[index] == n-k+index+1){ //如果这个位置上也达到最大值了,就需要往前一位
index--;
continue;
}
helper[index] += 1;
for(int i = index+1;i<k;i++) //初始化
helper[i] = helper[i-1]+1;
index = k-1;
}
vector<int> temp(helper,helper+k);
res.push_back(temp); //满足条件的数组
helper[index] += 1;
}else{
index--;
}
}
return res;
}
结果:
RuningTime | Memory | faster than |
---|---|---|
60 ms | 12.2 MB | 96.25% |