源码网商城,靠谱的源码在线交易网站 我的订单 购物车 帮助

源码网商城

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

  • 时间:2021-10-23 20:35 编辑: 来源: 阅读:
  • 扫一扫,手机访问
摘要:如何用itertools解决无序排列组合的问题
最近我作为Python菜鸟一枚开始征战[b]Codewars[/b],所以打算在这里记下遇到的有意思的题目。今天这第一题叫做“[b]Best Travel[/b]”: John和Mary计划去一些小镇旅行。Mary已经列好了这些小镇之间的距离比如ls=[50, 55, 57, 58, 60]。但是John不想开车太累,所以提出了两个要求:1) 开车不超过某个距离比如t=174 miles 2) 只能去3个小镇。 选择哪3个小镇可以让John和Mary都满意呢?(即找到距离之和最接近或等于t的3个小镇) [b]这道题目可以抽象成:[/b] 输入一个整数列表ls和整数t: 1. 找出从ls中任取三个元素的所有组合 2. 计算每个组合的三个元素之和 3. 如果存在小于或等于t的和,那么从中挑出最大的,然后输出这个最大和对应的三元素组合 4. 如果不存在,只好返回None [b]实现要点:[/b] [b]1. 无序排列组合:[/b] 用itertools模块的combinations方法 [b]2. 求和:[/b] 用sum函数 [b]3. 求最大值:[/b] 用max函数 [b]4. 捕获异常:[/b] 用try-except 借用这一题的[b]某个best solution[/b], 实现代码为:
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解决无序排列组合的问题就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持编程素材网。
  • 全部评论(0)
联系客服
客服电话:
400-000-3129
微信版

扫一扫进微信版
返回顶部