用树的思想来看这道题,以nums={1,2,2}为例,先取1子序列,沿着1,递归继续往后取,path不断叠加子序列,同时result也不断实时更新将path加入,到底部以后先pop_back()弹出,递归返回上一层,如果发现下一个循环中要取的元素和前一个取的相同,则路径重复,不能再取了,pop_back,继续返回上一层,以此类推,每个大路径的子路径都是一部回溯,直至回溯至根部,继续下一个大路径的递归然后回溯,即可得出结果。
回溯法思路是:搜索每一条路,每次回溯是对具体的一条路径而言的。对当前搜索路径下的的未探索区域进行搜索时,则可能会有两种情况:
当前未搜索区域满足结束条件,则保存当前路径并退出当前搜索;
当前未搜索区域需要继续搜索,则遍历当前所有可能的选择:如果该选择符合要求,则把当前选择加入当前的搜索路径中,并继续搜索新的未探索区域。
上面说的未搜索区域指的是搜索某条路径时的未搜索区域,而不是全局的未搜索区域。
backtrack :未探索区域中到达结束条件的所有可能路径,path 变量是保存的是一条路径,res 变量保存的是所有搜索到的路径。所以当「未探索区域满足结束条件」时,需要把 path 放到结果 res 中。
path.pop() 的作用?我们从始至终只用了一个变量 path,所以当对 path 增加一个选择并 backtrack 之后,需要清除当前的选择,防止影响其他路径的搜索。
看代码:
#include <iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
private:
vector<vector<int>> result;
vector<int> path;
public:
vector<vector<int>> subsetsWithDup(vector<int>& nums) {
sort(nums.begin(), nums.end());
backtrack(nums, 0);
return result;
}
void backtrack(vector<int>& nums, int start) {
if (start > nums.size())return;
result.push_back(path);
for (int i = start; i < nums.size(); i++) {
if (i > start && nums[i] == nums[i - 1]) continue;
path.push_back(nums[i]);
backtrack(nums, i + 1);
path.pop_back();
}
}
};
测试用例: nums = {1,2,3,2,2,2,4};
int main()
{
vector<int> nums = {1,2,3,2,2,2,4};
Solution* x = new Solution;
vector<vector<int>> result = x->subsetsWithDup(nums);
for (int i = 0; i < result.size(); i++) {
for (int j = 0; j < result[i].size(); j++) {
cout << result[i][j] << " ";
}
cout << endl;
}
system("pause");
}
测试结果: