用itertools解决无序排列组合问题

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了用itertools解决无序排列组合问题相关的知识,希望对你有一定的参考价值。

最近我作为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, 实现代码为:

1 def choose_best_sum(t, k, ls):
2     import itertools
3 
4     try:
5         return max(sum(combination) for combination in itertools.combinations(ls, k) if sum(combination) <= t)
6     except:
7         return None

 

以上是关于用itertools解决无序排列组合问题的主要内容,如果未能解决你的问题,请参考以下文章

Python3 - 排列组合的迭代

itertools---排列组合

如何指定仅允许某些第一个组合的 itertools 排列?

python itertools模块实现排列组合

什么是 Python itertools 的 Ruby 等价物,尤其是。组合/排列/分组?

Python Itertools 仅排列字母和数字