Python基于回溯法子集树模板解决数字组合问题实例
作者:罗兵 时间:2022-12-18 15:57:26
本文实例讲述了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程序设计有所帮助。
来源:http://www.cnblogs.com/hhh5460/p/6920382.html
标签:Python,回溯法
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
PHP实现设计模式中的抽象工厂模式详解
2024-06-05 09:39:58
![](https://img.aspxhome.com/file/2023/9/122529_0s.jpg)
Vue 项目中遇到的跨域问题及解决方法(后台php)
2023-11-16 09:14:13
![](https://img.aspxhome.com/file/2023/0/100540_0s.png)
什么是Python中的顺序表
2021-07-12 22:17:47
![](https://img.aspxhome.com/file/2023/8/115138_0s.png)
[译稿]Web 字体的未来
2008-07-31 17:50:00
python 装饰器的实际作用有哪些
2023-04-10 15:00:50
解决“引入同一个JS文件在非IE6中正常,但IE6报错”的问题
2009-04-03 11:42:00
详解pyqt5的UI中嵌入matplotlib图形并实时刷新(挖坑和填坑)
2023-01-04 22:01:05
Python实现基于socket的udp传输与接收功能详解
2023-09-04 03:23:20
![](https://img.aspxhome.com/file/2023/2/121412_0s.png)
SQL server高级应用 收藏版
2024-01-18 06:38:34
vue绑定class与行间样式style详解
2024-05-08 10:12:12
Python3.6简单反射操作示例
2021-05-04 03:32:05
![](https://img.aspxhome.com/file/2023/0/105900_0s.png)
java 中JDBC连接数据库代码和步骤详解及实例代码
2024-01-27 16:35:14
python的即时标记项目练习笔记
2022-11-25 05:49:35
![](https://img.aspxhome.com/file/2023/9/118499_0s.png)
golang使用json格式实现增删查改的实现示例
2024-04-27 15:36:51
python构造icmp echo请求和实现网络探测器功能代码分享
2023-07-14 15:53:43
Python3实现从文件中读取指定行的方法
2021-01-06 04:18:17
pandas数据清洗实现删除的项目实践
2021-09-02 16:44:17
Linux下实现C++操作Mysql数据库
2024-01-21 21:25:26
![](https://img.aspxhome.com/file/2023/0/115380_0s.png)
vue.js实现日历插件使用方法详解
2024-05-13 09:38:43
![](https://img.aspxhome.com/file/2023/8/125278_0s.jpg)
Sub-Pixel Bug?!
2010-03-24 18:09:00
![](https://img.aspxhome.com/file/UploadPic/20103/24/1749_001-51s.jpg)