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中使用SAX解析xml实例,利用SAX解析XML文档牵涉到两个部分:解析器和事件处理器,需要的朋友可以参考下2014-11-11
这篇文章主要给大家介绍了Python文件基本操作的相关资料，其中包括打开文件的方式、按行读取文件内容、复制文件、重命名文件等操作需要的朋友可以参考下2021-05-05
python中@property和property函数常见使用方法示例
这篇文章主要介绍了python中@property和property函数常见使用方法,结合实例形式分析了Python @property和property函数功能、使用方法及相关操作注意事项,需要的朋友可以参考下2019-10-10
今天小编就为大家分享一篇numpy 声明空数组详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-12-12
这篇文章主要介绍了使用 python 实现单人AI 扫雷游戏,今天我们用 Python 完成这个小程序，并且用AI来学习并实现它，需要的朋友可以参考下2021-08-08
下面小编就为大家带来一篇读写json中文ASCII乱码问题的解决方法。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2016-11-11
这篇文章主要为大家详细介绍了Python如何实现快速计算24点游戏并获取表达式，文中的示例代码讲解详细，感兴趣的小伙伴可以了解一下2022-12-12
这篇文章主要为大家详细介绍了如何利用Python开发一个自动答题程序，文中的示例代码讲解详细，感兴趣的小伙伴可以跟随小编一起了解一下2023-02-02
Python机器学习算法库scikit-learn学习之决策树实现方法详解
这篇文章主要介绍了Python机器学习算法库scikit-learn学习之决策树实现方法,结合实例形式分析了决策树算法的原理及使用sklearn库实现决策树的相关操作技巧,需要的朋友可以参考下2019-07-07
这篇文章主要为大家介绍了Python中Array特性与应用实例深入探究,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2024-01-01
最新评论