我们将使用回溯法的基本思想来解决子集问题,集合问题,排列问题,切割问题,棋盘问题。而回溯问题呢,都可以抽象成一个树形结构,我们在分析问题的时候,就需要画出相应的树形结构,最后判断收集结果的条件,以及符合条件的结果。回溯法的代码框架如下
void backtrack(参数列表)//回溯算法的参数很重要,
{
if(符合终止条件)
{
收集结果
//return ;//这个return有时可加有时不可加,具体看情况,看结果是树形结构的什么节点了
}
for(选择:本层集合中元素(树中节点孩子的数量就是集合的大小)
{
处理节点
backtrack(路径,选择列表); // 递归
回溯,撤销前面的结果
}
}
以本题为例,我们需要求一个集合元素互不相同的集合的子集。属于集合问题,我们就需要使用回溯算法。我们画出的树形结构如下:
我们可以知道,我们需要在每个节点都收集结果。而子集问题实际上和组合问题都很像,都是在一个集合上操作的,因此我们需要一个startindex来指示我们下一次递归开始的起始点。具体代码如下:
vector<vector<int>>res;
vector<int> path;
void backtrack(vector<int>&num,int startindex)
{
res.push_back(path);
if(startindex>=num.size())
{
return ;
}
for(int i=startindex;i<num.size();i++)
{
path.push_back(num[i]);
backtrack(num,i+1);
path.pop_back();
}
}
vector<vector<int>> subsets(vector<int>& nums)
{
backtrack(nums,0);
return res;
}