回溯算法是算法分析与设计中一种重要的算法设计思路,其本质就是类似暴力的方式去穷举可达到可行解的目标序列。
其主要的算法架构如下:
void backtrack(路径,选择列表)
{
if(满足结束条件)
{
result.push_back(结果);
}
for(选择:选择列表)
{
做出选择;
backtrack(路径,选择列表);
撤销选择;
}
}
结合leetcode 中的一道题目组合总数来讲解回溯算法
给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。
candidates 中的每个数字在每个组合中只能使用一次。
说明:
所有数字(包括目标数)都是正整数。
解集不能包含重复的组合。
输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
这道题不是入门级别的回溯算法题,因为其还涉及到减枝和去重复的过程,那么怎么减枝处理呢,其实我们在把数字添加到列表中的时候一旦列表中的数字超过了target的值,那么我们就直接返回就可以了。那么怎么去重复呢,首先我们对candidates数组进行排序处理,然后再转化为去除相同数字的问题,具体代码如下:
class Solution {
public:
vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
sort(candidates.begin(),candidates.end());
vector<vector<int>> v;
vector<int> result;
vector<int> flag(candidates.size(),0);
DFS(v,result,flag,target,candidates,0);
return v;
}
void DFS(vector<vector<int>>&v,vector<int> result,vector<int> flag,int target,vector<int> candidates,int index)
{
if(accumulate(result.begin(),result.end(),0)>target) //减枝
return;
if(accumulate(result.begin(),result.end(),0)==target)
{
v.push_back(result);
return;
}
for(int i=index;i<candidates.size();i++)
{
if(i>0&&candidates[i]==candidates[i-1]&&!flag[i-1]) //去重复
continue;
result.push_back(candidates[i]);
flag[i]=1;
DFS(v,result,flag,target,candidates,i+1);
result.pop_back();
flag[i]=0;
}
}
};