python 回溯法 子集树模板 系列 —— 12、选排问题

简介: 问题从n个元素中挑选m个元素进行排列,每个元素最多可重复r次。其中m∈[2,n],r∈[1,m]。如:从4个元素中挑选3个元素进行排列,每个元素最多可重复r次。分析解x的长度是固定的,为m。对于解x,先排第0个位置的元素x[0],再排第1个位置的元素x[1]。

问题

从n个元素中挑选m个元素进行排列,每个元素最多可重复r次。其中m∈[2,n],r∈[1,m]。

如:从4个元素中挑选3个元素进行排列,每个元素最多可重复r次。

分析

解x的长度是固定的,为m。

对于解x,先排第0个位置的元素x[0],再排第1个位置的元素x[1]。我们把后者看作是前者的一种状态,即x[1]是x[0]的一种状态!!

一般地,把x[k]看作x[k-1]的状态空间a中的一种状态,我们要做的就是遍历a[k-1]的所有状态。

那么,套用子集树模板即可。

代码

'''
选排问题

从n个元素中挑选m个元素进行排列,每个元素最多可重复r次。其中m∈[2,n],r∈[1,m]。

作者:hhh5460
时间:2017年6月2日 09时05分
声明:此算法版权归hhh5460所有

'''

n = 4
a = ['a','b','c','d']

m = 3   # 从4个中挑3个
r = 2   # 每个元素最多可重复2

x = [0]*m   # 一个解(m元0-1数组)
X = []      # 一组解

    
# 冲突检测
def conflict(k):
    global n, r, x, X, a
    
    # 部分解内的元素x[k]不能超过r
    if  x[:k+1].count(x[k]) > r:
        return True
        
    return False # 无冲突
    
    
# 用子集树模板实现选排问题
def perm(k): # 到达第k个元素
    global n,m, a, x, X
    
    if k == m:  # 超出最尾的元素
        print(x)
        #X.append(x[:]) # 保存(一个解)
    else:
        for i in a: # 遍历x[k-1]的状态空间a,其它的事情交给剪枝函数!
            x[k] = i
            if not conflict(k): # 剪枝
                perm(k+1)


# 测试
perm(0) # 从x[0]开始排列

效果图

img_5c511dac26dd1055fd21e8e13f6f4d1a.jpg

目录
相关文章
|
4月前
|
算法 Python
Python算法——树的遍历顺序变换
Python算法——树的遍历顺序变换
178 4
|
4月前
|
算法 搜索推荐 Python
Python算法——树的拓扑排序
Python算法——树的拓扑排序
70 0
|
11月前
|
算法 Python
Python递归树结构、回溯法实现
Python递归树结构、回溯法实现
|
Python 图形学 机器学习/深度学习
如何用Python画一棵漂亮的树
Tree 海龟绘图turtle 在1966年,Seymour Papert和Wally Feurzig发明了一种专门给儿童学习编程的语言——LOGO语言,它的特色就是通过编程指挥一个小海龟(turtle)在屏幕上绘图。
11518 0