如何用itertools解决无序排列组合的问题
最近我作为Python菜鸟一枚开始征战Codewars,所以打算在这里记下遇到的有意思的题目。今天这第一题叫做“BestTravel”:
John和Mary计划去一些小镇旅行。Mary已经列好了这些小镇之间的距离比如ls=[50,55,57,58,60]。但是John不想开车太累,所以提出了两个要求:1)开车不超过某个距离比如t=174miles2)只能去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
借用这一题的某个bestsolution,实现代码为:
defchoose_best_sum(t,k,ls): importitertools try: returnmax(sum(combination)forcombinationinitertools.combinations(ls,k)ifsum(combination)<=t) except: returnNone
以上这篇如何用itertools解决无序排列组合的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持毛票票。