题目:组合
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[[2,4],[3,4],[2,3],[1,2],[1,3],[1,4],]
思路1:backtrack回溯backtrack(index, n)
从当前位index到末位n遍历,将整数 i 加入现有组合,继续backtrack(i+1, n);
当剩余位不足选满k个数时枝剪。
代码如下:
class Solution {
public:
vector<vector<int>> res; //储存答案
void dfs(int index, int n, int k, vector<int>& temp){
//index为起始位,n为结束位,k代表还需选的个数,temp存储当前组合
if(k == 0){ //选满k个后把当前组合存入res
res.push_back(temp);
return;
}
for(int i = index; i <= n - k + 1; i++){ //枝剪
temp.push_back(i);
dfs(i + 1 , n, k - 1, temp);
temp.pop_back();
}
}
vector<vector<int>> combine(int n, int k) {
res.clear();
if(n * k == 0) return res;
vector<int> temp;
dfs(1, n, k, temp);
return res;
}
};
思路2:用全排列生成0、1序列作为选与不选的标识,再根据标识得到答案
代码如下:
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
vector<int> map(n,0);
vector<vector<int>> res;
for(int i=0;i<k;i++)
map[i]=1; //将map前k位置为1,作为初始情况,代表选择前k个数
vector<int> tmp;
do
{
for(int i=0;i<n;i++)
if(map[i]) tmp.push_back(i+1);
res.push_back(tmp);
tmp.clear();
}while(prev_permutation(map.begin(),map.end())); //生成比当前排列小的最大排列
return res;
}
};
大佬解法=.=:
class Solution {
public:
vector<vector<int>> combine(int n, int k) {
vector<vector<int>> a;
vector<int> b(k,0);
int i=0;
while (i >= 0)
{
b[i]++;
if (b[i] > n) i--;
else if (i == k - 1) a.push_back(b);
else
{
i++;
b[i] = b[i - 1];
}
}
return a;
}
};
作者:zhao-hu-2
链接:https://leetcode-cn.com/problems/combinations/solution/kan-de-ying-wen-ban-de-ti-jie-shi-zai-shi-shen-lai/
来源:力扣(LeetCode)