Python基于回溯法子集树模板解决数字组合问题实例
更新时间：2017年09月02日 12:32:52 作者：罗兵
这篇文章主要介绍了Python基于回溯法子集树模板解决数字组合问题,简单描述了数字组合问题并结合实例形式分析了Python回溯法子集树模板解决数字组合问题的具体步骤与相关操作技巧,需要的朋友可以参考下
本文实例讲述了Python基于回溯法子集树模板解决数字组合问题。分享给大家供大家参考，具体如下：
问题
找出从自然数1、2、3、...、n中任取r个数的所有组合。
例如，n=5，r=3的所有组合为：
1,2,3
1,2,4
1,2,5
1,3,4
1,3,5
1,4,5
2,3,4
2,3,5
2,4,5
3,4,5
分析
换个角度，r=3的所有组合，相当于元素个数为3的所有子集。因此，在遍历子集树的时候，对元素个数不为3的子树剪枝即可。
注意，这里不妨使用固定长度的解。
直接套用子集树模板。
代码
'''数字组合问题'''
n = 5
r = 3
a = [1,2,3,4,5] # 五个数字
x = [0]*n # 一个解（n元0,1数组） 固定长度
X = [] # 一组解
def conflict(k):
global n, r, x
if sum(x[:k+1]) > r: # 部分解的长度超出r
return True
if sum(x[:k+1]) + (n-k-1) < r: # 部分解的长度加上剩下长度不够r
return True
return False # 无冲突
# 套用子集树模板
def comb(k): # 到达第k个元素
global n, x, X
if k >= n: # 超出最尾的元素
#print(x)
X.append(x[:]) # 保存（一个解）
else:
for i in [1, 0]: # 遍历元素 a[k] 的两种选择状态:1-选择，0-不选
x[k] = i
if not conflict(k): # 剪枝
comb(k+1)
# 根据一个解x，构造对应的一个组合
def get_a_comb(x):
global a
return [y[0] for y in filter(lambda s:s[1]==1, zip(a, x))]
# 根据一组解X，构造对应的一组组合
def get_all_combs(X):
return [get_a_comb(x) for x in X]
# 测试
comb(0)
print(X)
print(get_all_combs(X))
效果图
更多关于Python相关内容感兴趣的读者可查看本站专题：《Python数据结构与算法教程》、《Python函数使用技巧总结》、《Python字符串操作技巧汇总》及《Python入门与进阶经典教程》
希望本文所述对大家Python程序设计有所帮助。
相关文章
这篇文章主要介绍了Python实现两个list求交集，并集，差集的方法,结合实例形式分析了Python使用intersection、union及difference方法实现两个集合list的交集、并集与差集操作技巧,需要的朋友可以参考下2018-08-08
这篇文章主要介绍了python递归&迭代方法实现链表反转，文章分享一段详细实现代码，需要的小伙伴可以参考一下，希望对你的学习或工作有所帮助2022-02-02
python中的selenium安装的步骤（浏览器自动化测试框架）
这篇文章主要介绍了python中的selenium安装的步骤，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-03-03
这篇文章主要介绍了python简单区块链模拟详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了python executemany的使用及注意事项,非常不错，具有参考借鉴价值，需要的朋友可以参考下2017-03-03
这篇文章主要为大家介绍了Python基础之文件操作及光标移动详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-11-11
well,我们这里所指的文件打开方式并不是指调用什么应用程序去打开某个文件,而是只读只写或者二进制等的打开方式,这里我们就来详解Python中open()函数指定文件打开方式的用法2016-06-06
conda管理包很方便,但并不是所有包都有最新的conda版本,所以仍需要使用conda中的pip来安装包,下面这篇文章主要给大家介绍了关于conda虚拟环境使用pip下载包到当前环境的两种方法,需要的朋友可以参考下2022-08-08
这篇文章主要介绍了python实现单例的两种方法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-03-03
这篇文章主要介绍了Python如何读写CSV文件，帮助大家更好的理解和学习python，感兴趣的朋友可以了解下2020-08-08
最新评论