Python回溯法子集树模板：解决元素选排与重复限制问题

雨后青山

发布时间：2026-10-13| 2人浏览过

原创

本文介绍利用Python回溯法结合子集树模板解决选排问题的方法。针对从n个元素中选取m个进行排列且每个元素最多重复r次的约束条件，通过定义固定长度解空间并逐位遍历状态，配合冲突检测剪枝函数，高效生成所有合法排列。

本文分享一个使用Python回溯法结合子集树模板来解决选排问题的具体实例，供读者参考学习。

问题描述

核心任务是从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[1]是x[0]状态空间中的一种具体表现。

推广到一般情况，将x[k]视为x[k-1]状态空间a中的一种状态，我们的目标就是遍历x[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]开始排列

运行效果展示