1 回溯法
讲真,这道题加入重复元素,还是有点难度的
具体策略为
- 排序使重复元素相邻
- 记录已用元素避免重复(
inPath
) - 跳过重复元素(
nums[i-1] == nums[i] && !inPath[i-1]
)
class Solution {
private:
int size;
vector<int> path; // 路径
vector<vector<int>> solution;
public:
void backtrack(vector<int>& nums, vector<bool>& inPath) {
if (path.size() == size) {
solution.emplace_back(path);
return;
}
for (int i = 0; i < size; i ) {
// 3.跳过重复元素,两者相同且nums[i-1]用过则nums[i]不再用;
if (i > 0 && nums[i-1] == nums[i] && !inPath[i-1]) continue;
if (inPath[i]) continue;
inPath[i] = true;
path.emplace_back(nums[i]);
backtrack(nums, inPath);
path.pop_back();
inPath[i] = false; // 若nums[i] == nums[i 1], 则需inPath[i] == false辅助判断, 表用过
}
}
vector<vector<int>> permuteUnique(vector<int>& nums) {
size = nums.size();
// 1.升序排序(使重复元素相邻)
sort(nums.begin(), nums.end());
// 2.记录已用元素避免重复
vector<bool> inPath(size, false); // 是否在路径中
backtrack(nums, inPath);
return solution;
}
};
其中,以下两段代码都可以通过,但inPath[i-1] == false
效率更高
if (i > 0 && nums[i-1] == nums[i] && !inPath[i-1]) continue;
if (i > 0 && nums[i-1] == nums[i] && inPath[i-1]) continue;
具体见图
致谢
感谢「代码随想录」公众号梳理的思路,欢迎大家关注这位大佬的公号