如何用itertools解決無序排列組合的問題
最近我作為Python菜鳥一枚開始征戰(zhàn)Codewars，所以打算在這里記下遇到的有意思的題目。今天這第一題叫做“Best Travel”：
John和Mary計劃去一些小鎮(zhèn)旅行。Mary已經(jīng)列好了這些小鎮(zhèn)之間的距離比如ls=[50, 55, 57, 58, 60]。但是John不想開車太累，所以提出了兩個要求：1) 開車不超過某個距離比如t=174 miles 2) 只能去3個小鎮(zhèn)。
選擇哪3個小鎮(zhèn)可以讓John和Mary都滿意呢？（即找到距離之和最接近或等于t的3個小鎮(zhèn)）
這道題目可以抽象成：
輸入一個整數(shù)列表ls和整數(shù)t：
1. 找出從ls中任取三個元素的所有組合
2. 計算每個組合的三個元素之和
3. 如果存在小于或等于t的和，那么從中挑出最大的，然后輸出這個最大和對應(yīng)的三元素組合
4. 如果不存在，只好返回None
實現(xiàn)要點：
1. 無序排列組合：
用itertools模塊的combinations方法
2. 求和：
用sum函數(shù)
3. 求最大值：
用max函數(shù)
4. 捕獲異常：
用try-except
借用這一題的某個best solution, 實現(xiàn)代碼為：
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解決無序排列組合的問題就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了Python matplotlib可視化實例解析,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-06-06
這篇文章主要介紹了使用python實現(xiàn)3D聚類圖效果,本文通過實例代碼給大家介紹的非常詳細(xì),感興趣的朋友跟隨小編一起看看吧2024-08-08
python利用Appium實現(xiàn)自動控制移動設(shè)備并提取數(shù)據(jù)功能
這篇文章主要介紹了python利用Appium自動控制移動設(shè)備并提取數(shù)據(jù),本文以控制抖音app滑動并獲取抖音短視頻發(fā)布者為例，通過實例代碼給大家介紹的非常詳細(xì)，需要的朋友可以參考下2021-09-09
python實現(xiàn)批量監(jiān)聽頁面并發(fā)送郵件
這篇文章主要為大家詳細(xì)介紹了python如何實現(xiàn)自動化批量監(jiān)聽頁面并發(fā)送郵件,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2023-11-11
python?matplotlib繪畫十一種常見數(shù)據(jù)分析圖
這篇文章主要介紹了python?matplotlib繪畫十一種常見數(shù)據(jù)分析圖，文章主要繪制折線圖、散點圖、直方圖、餅圖等需要的小伙伴可以參考一下文章具體內(nèi)容2022-06-06
這篇文章主要介紹了PyQt5打開文件對話框QFileDialog實例代碼，分享了相關(guān)代碼示例，小編覺得還是挺不錯的，具有一定借鑒價值，需要的朋友可以參考下2018-02-02
這篇文章主要介紹了python 繪制場景熱力圖的示例，幫助大家更好的利用python繪制圖像，感興趣的朋友可以了解下2020-09-09
這篇文章主要介紹了如何在python字符串中輸入純粹的{}以及python字符串連接的三種方法,需要的朋友可以參考下2018-08-08
這篇文章主要介紹了Python高階函數(shù)的相關(guān)資料，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下2020-08-08
這篇文章主要介紹了Python淺拷貝與深拷貝用法,實例分析了Python淺拷貝與深拷貝的功能與使用方法,需要的朋友可以參考下2015-05-05