如何用itertools解决无序排列组合的问题
2017-05-18 07:51
495 查看
最近我作为Python菜鸟一枚开始征战Codewars,所以打算在这里记下遇到的有意思的题目。今天这第一题叫做“Best Travel”:
John和Mary计划去一些小镇旅行。Mary已经列好了这些小镇之间的距离比如ls=[50, 55, 57, 58, 60]。但是John不想开车太累,所以提出了两个要求:1) 开车不超过某个距离比如t=174 miles 2) 只能去3个小镇。
选择哪3个小镇可以让John和Mary都满意呢?(即找到距离之和最接近或等于t的3个小镇)
这道题目可以抽象成:
输入一个整数列表ls和整数t:
1. 找出从ls中任取三个元素的所有组合
2. 计算每个组合的三个元素之和
3. 如果存在小于或等于t的和,那么从中挑出最大的,然后输出这个最大和对应的三元素组合
4. 如果不存在,只好返回None
实现要点:
1. 无序排列组合:
用itertools模块的combinations方法
2. 求和:
用sum函数
3. 求最大值:
用max函数
4. 捕获异常:
用try-except
借用这一题的某个best solution, 实现代码为:
def choose_best_sum(t, k, ls): import itertools try: return max(sum(combination) for combination in itertools.combinations(ls, k) if sum(combination) <= t) except: return None
以上这篇如何用itertools解决无序排列组合的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。
您可能感兴趣的文章:
相关文章推荐
- C语言实现的排列组合问题的通用算法、解决方法
- python 排列组合之itertools
- 引入进制概念解决排列组合问题
- Python itertools 排列组合
- 排列平方数 若干不同的数字,排列组合后能产生多少个平方数? 下面的代码解决了这个问题。
- python itertools的函数(含有排列组合函数)
- Java解决排列组合问题——深度优先遍历
- 程序员的数学 - 排列组合 - 解决计数问题的方法 - 阅读小结
- python 排列组合 解决实际问题
- python 排列组合之itertools
- 如何解决win10桌面图标位置和排列方式无法保存问题
- Java解决排列组合问题——深度优先遍历
- AC解 - 用动态规划解决一道排列组合计数问题(序关系计算)
- 解决组合排列问题 A (m ,n) m>=n
- 如何解决JsonObject无序的问题
- 【Python】排列组合itertools & 集合set
- python 排列组合之itertools
- Java解决排列组合问题——深度优先遍历
- Java解决排列组合问题——深度优先遍历
- Java解决排列组合问题——深度优先遍历