给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。candidates 中的数字可以无限制重复被选取。
说明:
- 所有数字(包括 target)都是正整数。
- 解集不能包含重复的组合。
示例 1:
输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
[7],
[2,2,3]
]
解题思路:使用递归+回溯,为了避免组合出现重复,例如示例1中的重复组合[2,2,3]和[3,2,2],可以先对输入candidates进行排序,然后在递归的时候,设temp为当前组合,下一个可以放进temp中的数,一定大于等于temp[-1]。
其Python3代码如下:
# Python3
class Solution:
def __init__(self):
self.ans = []
def combinationSum(self, candidates: List[int], target: int) -> List[List[int]]:
if not candidates:
return []
def combSums(candidates,target,temp,sums):
if target == sums: # 递归停止条件
self.ans.append(temp[:])
return
for n in candidates: # 遍历每种情况
if sums + n > target: # 如果当前sums值大于target,直接break
break
if temp and n < temp[-1]: # 如果当前输入的值小于temp[-1]则continue
continue
temp.append(n)
combSums(candidates,target,temp,sums+n)
temp.pop() # 回溯
candidates.sort() # 执行递归时先进行排序
combSums(candidates,target,[],0)
return self.ans
其C++代码如下:
# C++
class Solution {
private:
vector<vector<int>> num;
vector<int> sum;
int i = 0;
public:
vector<vector<int>> combinationSum(vector<int>& candidates, int target)
{
sort(candidates.begin(),candidates.end());
function(candidates,target,sum,i);
return num;
}
void function(vector<int>& candidate, int target,vector<int> sum,int i)
{
if(target==0)
{
num.push_back(sum);
}
for(;i<candidate.size();++i) # 只遍历比当前数字大的candidates数字
{
if(candidate[i]<=target)
{
sum.push_back(candidate[i]);
function(candidate,target-candidate[i],sum,i);
sum.pop_back();
}
}
}
};