python之排列组合1

啥也不懂的人学编程2024-09-17 22:57

排列组合

不重复全排列

给定一个可包含重复数字的序列nums,按任意顺序返回所有不重复的全排列

示例1：

输入：nums= $1,1,2$

输出： $\[1,1,2$ , $1,2,1$ , $2,1,1$ ]

输入：nums= $1,2,3$

输出： $\[1,2,3$ , $1,3,2$ , $2,1,3$ , $2,3,1$ , $3,1,2$ , $3,2,1$ ]

思路：

1，对所有的列表元素进行全排列

2，删除重复排列的元素

复制代码

from typing import List  # 导入列List模块


class Solution:  # 建立类
    def combination(self, nums: List[int]):  # 定义函数，并定义输入格式
        if len(nums) == 0:  # 如果为空列表则返回空[]
            return []
        elif len(nums) == 1:  # 如果列表只有一个项，则返回列表本身
            return [nums]  # 一定要用[nums]-返回列表，return nums则是返回列表的value
        else:
            ans = []  # 建立空列表，用于存储中间值
            for index, item in enumerate(nums):  # 遍历列表的索引及值
                rel = nums[:index] + nums[index + 1:]  # 提取除了当前项的以外的nums列表
                print('index:', index, 'item:', item)  # 打印输出索引及值，此输出为中间步骤输出方便理解，可以注释掉
                print('rel:', rel)  # 打印剔除当前项以外的列表，此输出为中间步骤输出方便理解，可以注释掉
                for j in self.combination(rel):  # 递归调用，
                    ans.append(j + [item])  # 形成新的组合
                    print(ans)  # 此输出为中间步骤输出方便理解，可以注释掉
            result = []  # 输出结果的空列表
            for i in ans:  # 遍历ans中间存储列表
                if i not in result:  # 去重，不同排列的组合存在result列表中，重复项目不保存
                    result.append(i)
            return result


list_1 = []
s = Solution()
print(s.combination(list_1))
print('----1')
list_1 = [1, 1]
print(s.combination(list_1))
print('----2')
list_1 = [1, 1, 2]
print(s.combination(list_1))
print('----3')