2021-09-04

全排列问题–有重复元素、去重

# !/usr/bin/env python
# -*- coding: utf-8 -*-

# # nums含重复元素的时候，使用record存储每层已经选过的元素，好理解！缺点：额外空间复杂度高！
# class Solution:
#     def permute(self, nums, i):
#         if i == len(nums):
#             result.append(nums[:])
#             return
#         record = set()  # 创建record集合用来存储该递归层即第i个元素已经选择过的数字，用来去重
#         for j in range(i, len(nums)):  # nums[i...end]中可能有重复元素，第i个元素取值时不能取到重复元素
#             if nums[j] not in record:
#                 record.add(nums[j])
#                 nums[i], nums[j] = nums[j], nums[i]
#                 self.permute(nums, i+1)
#                 nums[i], nums[j] = nums[j], nums[i]
#
#
# if __name__ == '__main__':
#     result = list()
#     nums = [1, 1, 2]
#     p = Solution()
#     p.permute(nums, 0)
#     print(result)


# 不在每一次递归时创建record集合去记录该次递归已经选过的元素，那么就先给nums排序，重复元素必定相邻！！！
class Solution:
    def permute(self, nums):  # 此时不能在nums中互换i，j位置，因为需要排序后判断[j]和[j-1]！！！
        if len(path) == len(nums):  # 所以需要path和vis数组
            result.append(path[:])
            return
        for j in range(len(nums)):
            if (j > 0 and nums[j] == nums[j-1]) and vis[j-1] == 0:  
                # 当nums[j] == nums[j-1]并且vis[j-1] == 0，则表明对于nums[j]来说已经在该层递归中选过了，因为是顺序遍历选择，所以如果nums[j-1]没有选过，肯定会先选nums[j-1]
                # 因为如果此时vis[j-1] == 1，则表明该层的上一层递归还未结束，该层可以与上一层选择一样的元素，不算重复排列
                continue
            if vis[j] == 0:
                vis[j] = 1
                path.append(nums[j])
                self.permute(nums)
                path.pop()
                vis[j] = 0


if __name__ == '__main__':
    result = list()
    path = list()
    nums = [1, 1, 2]
    nums.sort()  # 先排序！！！
    vis = [0] * len(nums)
    p = Solution()
    p.permute(nums)
    print(result)