LeetCode算法题47:全排列 II解析

给定一个可包含重复数字的序列,返回所有不重复的全排列。

示例:

输入: [1,1,2]
输出:
[
  [1,1,2],
  [1,2,1],
  [2,1,1]
]

这个题和上一个题全排列是一样的,只是在判断条件中在加一些防止重复元素重复出现的元素就可以了。这里加到最里面。

C++源代码:

class Solution {
public:
    vector<vector<int>> permuteUnique(vector<int>& nums) {
        vector<vector<int>> res;
        vector<int> out;
        vector<int> flag(nums.size(), 0);
        sort(nums.begin(), nums.end());
        permuteUniqueDFS(nums, 0, flag, out, res);
        return res;
    }
    void permuteUniqueDFS(vector<int>& nums, int layer, vector<int>& flag, vector<int>& out, vector<vector<int>>& res)
    {
        if(layer>=nums.size()) res.push_back(out);
        else
        {
            for(int i=0;i<nums.size();i++)
            {
                if(flag[i]==0)
                {
                    if(i>0 && nums[i]==nums[i-1] && flag[i-1]==0) continue;
                    out.push_back(nums[i]);
                    flag[i] = 1;
                    permuteUniqueDFS(nums, layer+1, flag, out, res);
                    out.pop_back();
                    flag[i] = 0;
                }
            }
        }
    }
};

python3源代码:

class Solution:
    def permuteUnique(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        res = []
        out = []
        flag = [0 for i in range(len(nums))]
        nums.sort()
        self.permuteUniqueDFS(nums, 0, flag, out, res)
        return res
    
    def permuteUniqueDFS(self, nums, layer, flag, out, res):
        if layer >= len(nums):
            res.append(copy.deepcopy(out))
        else:
            for i in range(len(nums)):
                if flag[i] == 0:
                    if i>0 and nums[i]==nums[i-1] and flag[i-1]==0:
                        continue
                    out.append(nums[i])
                    flag[i] = 1
                    self.permuteUniqueDFS(nums, layer+1, flag, out, res)
                    out.pop()
                    flag[i] = 0

猜你喜欢

转载自blog.csdn.net/x603560617/article/details/85292750