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程序设计有所帮助。
相关文章
这篇文章主要介绍了Apache部署Django项目图文详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07
Pyecharts是一个由百度开源的、用于生成Echarts图表的类库，可以用来进行数据可视化分析。本文将详细讲解一下Pyecharts的使用，需要的可以参考一下2022-04-04
这篇文章主要介绍了Pandas DataFrame求差集的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-12-12
当你需要在Python中处理文本数据时,正则表达式绝对是你的瑞士军刀,今天小编就来和大家一起聊聊Python中re模块的那些实用技巧和常见陷阱吧2025-04-04
这篇文章主要介绍了Django url 路由匹配过程详解，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2021-01-01
这篇文章主要介绍了一些Centos Python 生产环境的部署命令，非常不错，具有参考借鉴价值，需要的朋友参考下吧2018-05-05
这篇文章主要介绍了python深度学习中的4种激活函数,帮助大家更好的进行深度学习，感兴趣的朋友可以了解下2020-09-09
这篇文章主要为大家详细介绍了plotly分割显示mnist的方法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助2022-03-03
这篇文章主要介绍了自己使用总结Python程序代码片段,本文收集了如反向读取文件、往文件中所有添加指定的前缀、匿名函数作为返回值、将二进制数转为10进制数等实用代码片段,需要的朋友可以参考下2015-06-06
ubuntu安装sublime3并配置python3环境的方法
这篇文章主要介绍了ubuntu安装sublime3并配置python3环境的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-03-03
最新评论