原题地址:https://leetcode-cn.com/problems/combinations/description/
题目描述:
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
解题方案:
之前遇到和本题类似的题目,都属于回溯算法的题目。记住回溯算法解题方式,以后套用就行了。回溯算法都采用的是dsf深度遍历的方式,遍历整个数组,情况1:如果当前为空(即添加当前组合的第一个元素) 或者 当前要添加的元素满足 大于 当前组合数组中的最后一个元素大小(因为该数组为升序),即可添加 。情况2:在情况1之前(即递归出口),如果当前组合的个数等于k,则添加到总的数组中去。每次进行深度遍历时,插入当前元素,遍历结束后弹出插入的元素,这种方式是dsf深度遍历的精髓。其他回溯算法的题目中,是在递归前进行修改,递归结束后进行恢复。
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;
}
};