Leetcode|排列|47. 全排列 II(排序使重复元素相邻+记录已用元素避免重复+跳过重复元素)

2021-09-18 16:11:04 浏览数 (1)

1 回溯法

讲真,这道题加入重复元素,还是有点难度的

具体策略为

  • 排序使重复元素相邻
  • 记录已用元素避免重复(inPath)
  • 跳过重复元素(nums[i-1] == nums[i] && !inPath[i-1])
代码语言:javascript复制
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效率更高

代码语言:javascript复制
if (i > 0 && nums[i-1] == nums[i] && !inPath[i-1])  continue;
if (i > 0 && nums[i-1] == nums[i] && inPath[i-1])  continue;

具体见图

致谢

感谢「代码随想录」公众号梳理的思路,欢迎大家关注这位大佬的公号

0 人点赞