小黑晚上又没法练琴了，于是去朝阳公园独自跑步吃了两份烤冷面喝了鸡尾酒的leetcode之旅:90. 子集 II

本文链接：https://blog.csdn.net/qq_37418807/article/details/129855874

小黑代码1

class Solution:
    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        # 对数组进行排序，以便回溯
        nums = sorted(nums)
        # 数组长度
        n = len(nums)
        # 结果数组与中间变量
        results = []
        res = []
        # 访问数组
        flags = [False] * n
        # 回溯函数
        def dfs(i = 0):
            # 递归出口
            if i == n:
                results.append(res[:])
                return
            
            # 没她
            dfs(i + 1)
            # 判断该元素在此位置是否被用过
            if i > 0 and nums[i-1] == nums[i] and (not flags[i-1]):
                return

            # 有她
            res.append(nums[i])
            flags[i] = True
            dfs(i + 1)
                # 回溯
            res.pop()
            flags[i] = False
        dfs()
        return results

在这里插入图片描述

小黑代码2

class Solution:
    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        # 对数组进行频率统计
        items = list(collections.Counter(nums).items())
        # 集合长度
        n = len(items)
        # 中间变量和结果数组
        results = []
        res = []
        # 回溯函数
        def dfs(i = 0):
            if i == n:
                results.append(res[:])
                return
            # 跳过该元素
            dfs(i + 1)
            # 不跳过
            count = items[i][1]
            for j in range(count):
                res.append(items[i][0])
                dfs(i + 1)
            for j in range(count):
                res.pop()
        dfs()
        return results

在这里插入图片描述

迭代法子集枚举

class Solution:
    def subsetsWithDup(self, nums: List[int]) -> List[List[int]]:
        # 对数组进行排序
        nums = sorted(nums)
        # 数组长度
        n = len(nums)
        # 结果数组
        results = []
        # 开始枚举
        for mask in range(1 << n):
            # 中间变量
            res = []
            # 标记
            flag = True
            # 匹配元素
            for i in range(n):
                if (mask >> i) & 1:
                    # 判断上一个元素是否被用过
                    if i > 0 and nums[i-1] == nums[i] and not (mask >> (i-1) & 1):
                        flag = False
                        break
                    res.append(nums[i])
            if flag:
                results.append(res)
        return results