思路:
注意点1:集合里含有重复元素,需要对元素进行去重,定义一个set集合,在遍历的时候对本层元素进行去重。不能先排序再去重,这样集合的次序就被打乱了。
注意点2:由于子序列需要是递增的,因此在本层新加入path数组元素时需要和前一个元素进行比较判断,if(!path.empty() && nums[i] < path.back()) continue;
代码:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, int startIndex){
if(path.size() > 1){
result.push_back(path);
//注意不要加return,因为要取树上的节点不是叶子节点
}
unordered_set<int> uset;//定义set集合对本层元素进行去重
for(int i = startIndex; i < nums.size(); i++){
if(!path.empty() && nums[i] < path.back() || uset.find(nums[i]) != uset.end() ){//如果新加入path数组的元素小于path数组末尾元素,说明不是递增序列,则跳过
//本次循环;如果新加入path数组的元素和set集合里的元素重复,则需要进行去重操作,跳过本次循环。
continue;
}
path.push_back(nums[i]);
uset.insert(nums[i]);//记录这个元素在本层已经使用过了,本层后面不能再用了
backtracking(nums, i + 1);
path.pop_back();
}
}
vector<vector<int>> findSubsequences(vector<int>& nums) {
result.clear();
path.clear();
if(nums.size() == 1) return result;
backtracking(nums, 0);
return result;
}
};
思路:
注意点1:排列问题和组合、子集、分割问题不同,在本层循环时i都从0开始遍历,因此没有参数startIndex来定位下一层循环的遍历起点。
注意点2:需要一个used数组来记录本层元素是否使用过,定义为vector<bool> used(nums.size(), false),当在本层使用过后,used[i] 标记为true,回溯时path数组和used数组要一起回溯。
代码:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
//参数used数组用来记录nums数组里的元素是否使用过
void backtracking(vector<int>& nums, vector<bool>& used){
//终止条件是path数组和nums数组大小相等,即来到了叶子节点
if(path.size() == nums.size()){
result.push_back(path);
return;
}
for(int i = 0; i < nums.size(); i++){//注意i从0开始遍历,没有startIndex
if(used[i] == true) continue;//在本层遍历时,如果前面已经使用过,直接跳过
used[i] = true;//元素使用过后,标记成true
path.push_back(nums[i]);
backtracking(nums, used);
path.pop_back();//path数组元素回溯
used[i] = false;//used数组回溯
}
}
vector<vector<int>> permute(vector<int>& nums) {
result.clear();
path.clear();
vector<bool> used(nums.size(), false);//used数组初始化大小等于nums数组大小,值都为false
backtracking(nums, used);
return result;
}
};
思路:
注意点:集合nums包含重复元素,需要进行去重。先对nums进行排序,让相同元素紧挨着一起。需要used数组用来记录元素是否使用过,再对本层遍历时进行去重操作:
对同一树层进行去重操作: if(i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false) continue;
对同一树枝进行去重操作: if(i > 0 && nums[i] == nums[i - 1] && used[i - 1] == true) continue;
代码:
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void backtracking(vector<int>& nums, vector<bool>& used){
if(path.size() == nums.size()){
result.push_back(path);
return;
}
for(int i = 0; i < nums.size(); i++){
// used[i - 1] == true,说明同一树枝nums[i - 1]使用过
// used[i - 1] == false,说明同一树层nums[i - 1]使用过
// 如果同一树层nums[i - 1]使用过则直接跳过
if(i > 0 && nums[i] == nums[i - 1] && used[i - 1] == false){
continue;
}
if(used[i] == false){
used[i] = true;
path.push_back(nums[i]);
backtracking(nums, used);
path.pop_back();
used[i] = false;
}
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
result.clear();
path.clear();
vector<bool> used(nums.size(), false);
sort(nums.begin(), nums.end());
backtracking(nums, used);
return result;
}
};