python3 排列组合

亦凉 2024-04-20 10:25 56阅读 0赞

**运行环境 : python 3.6.0**

**一 、python 自身内置就有这种工具 ，模块名为 itemtools 。**

**所有结果 :**

# -*- encoding=utf-8 -*-
    from itertools import product
    
    test_data = list('234')
    
    """ 所有结果 """
    for item in product(test_data, repeat=2):
        print(item)
    
    # 输出
    """
    ('2', '2')
    ('2', '3')
    ('2', '4')
    ('3', '2')
    ('3', '3')
    ('3', '4')
    ('4', '2')
    ('4', '3')
    ('4', '4')
    """

**排列 :**

# -*- encoding=utf-8 -*-
    from itertools import permutations
    
    test_data = list('234')
    
    """ 排列 """
    for item in permutations(test_data, 2):
        print(item)
    
    # 输出
    """
    ('2', '3')
    ('2', '4')
    ('3', '2')
    ('3', '4')
    ('4', '2')
    ('4', '3')
    """

**组合 :**

# -*- encoding=utf-8 -*-
    from itertools import combinations
    
    test_data = list('1234')
    
    """ 组合 """
    for item in combinations(test_data, 3):
        print(item)
    
    
    # 输出
    """
    ('1', '2', '3')
    ('1', '2', '4')
    ('1', '3', '4')
    ('2', '3', '4')
    """

**有替换的组合 ：**

# -*- encoding=utf-8 -*-
    from itertools import combinations_with_replacement
    
    test_data = list('234')
    
    """ 有替换的组合 """
    for item in combinations_with_replacement(test_data, 2):
        print(item)
    
    # 输出
    """
    ('2', '2')
    ('2', '3')
    ('2', '4')
    ('3', '3')
    ('3', '4')
    ('4', '4')
    """

**二 、自己设计算法也可以实现排列组合效果**

**组合 :**

import copy  # 实现list的深复制
    
    
    def combine(lst, k):
        """ lst: 数组  k: 组合长度  return: 组合 """
        result = []
        tmp = [0] * k
        length = len(lst)
    
        def next_num(Li=0, Ni=0):
            if Ni == k:
                result.append(copy.copy(tmp))
                return
            for Lj in range(Li, length):
                tmp[Ni] = lst[Lj]
                next_num(Lj + 1, Ni + 1)
    
        next_num()
        return result
    
    
    if __name__ == '__main__':
        lst = list('2346')
        for item in combine(lst, 3):
            print(item)