题目一:
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
示例 2:
输入:n = 1, k = 1 输出:[[1]]
思路:
一个组合固定要有k个元素,如果只按for循环,需要o(n**k)时间复杂度 明显不行,考虑使用递归的方法遍历是否由可行的组合
递归的参数和返回值: n、k和当前的元素值cur,当前的组合为curVec,最后的总组合ret,返回值可为空
递归终止条件:cur>n || curVec.size()==k
递归的行为: 用一个for循环遍历当前可能的组合
for(int i=cur;i<=n&&i<cur+k+1;i++)
{
curVec.push_back(i);
dfs(n,k,i+1,curVec,ret);
curVec.pop_back();
}
整体代码:
class Solution {
public:
void dfs(int n,int k,int cur,vector<int>& curVec,vector<vector<int>>& ret)
{
if(cur>n || curVec.size()==k)
{
if(curVec.size()==k) ret.push_back(curVec);
return;
}
for(int i=cur;i<=n&&i<cur+k+1;i++)
{
curVec.push_back(i);
dfs(n,k,i+1,curVec,ret);
curVec.pop_back();
}
return;
}
vector<vector<int>> combine(int n, int k) {
vector<int> curVec;
vector<vector<int>> ret;
dfs(n,k,1,curVec,ret);
return ret;
}
};
题目二:
找出所有相加之和为 n
的 k
个数的组合,且满足下列条件:
- 只使用数字1到9
- 每个数字 最多使用一次
返回 所有可能的有效组合的列表 。该列表不能包含相同的组合两次,组合可以以任何顺序返回。
示例 1:
输入: k = 3, n = 7 输出: [[1,2,4]] 解释: 1 + 2 + 4 = 7 没有其他符合的组合了。
示例 2:
输入: k = 3, n = 9 输出: [[1,2,6], [1,3,5], [2,3,4]] 解释: 1 + 2 + 6 = 9 1 + 3 + 5 = 9 2 + 3 + 4 = 9 没有其他符合的组合了。
思路:
与上一题基本类似,不同的是该题可加入组合的数字只能是1-9,且不可重复,则循环范围应该改为curIndex(当前数字)并且用一个set保存curVec已有数字,用来判断数字是否重复。
细节:同时维护遍历的为递增的Vec,可以省去对最后ret结果去重
if(curVec.size()>0 && i < curVec[curVec.size()-1]) continue;
class Solution {
public:
void dfs(int k,int n,int curIndex,int curSum,vector<int>& curVec,set<int>& curSet,vector<vector<int>>& ret)
{
//n超过当前k个数的组合上限,则return
if(curVec.size()==k)
{
if(curSum==n) ret.push_back(curVec);
return;
}
else if(curVec.size()>k || curSum>n) return;
for(int i=curIndex;i<=9;i++)
{
//如果curVec中已经有这个数了,则跳过
//建立递增的vec(用来去重)
if(curSet.count(i) || (curVec.size()>0 && i < curVec[curVec.size()-1])) continue;
curVec.push_back(i);
curSet.insert(i);
dfs(k,n,curIndex+1,curSum+i,curVec,curSet,ret);
curVec.pop_back();
curSet.erase(i);
}
return;
}
vector<vector<int>> combinationSum3(int k, int n) {
//需要的个数k,目标n,已加入组合的curVec,
vector<int> curVec;
set<int> curSet;
vector<vector<int>> ret;
dfs(k,n,1,0,curVec,curSet,ret);
return ret;
}
};