力扣 leetcode 561. 陣列拆分 I (python)
阿新 • • 發佈:2021-02-20
技術標籤:pythonleetcodeleetcode演算法python資料結構
Topic
給定長度為 2n 的整數陣列 nums ,你的任務是將這些數分成 n 對, 例如 (a1, b1), (a2, b2), …, (an, bn) ,使得從 1 到 n 的 min(ai, bi) 總和最大。
返回該 最大總和 。
Example_1
輸入:nums = [1,4,3,2]
輸出:4
解釋:所有可能的分法(忽略元素順序)為:
- (1, 4), (2, 3) -> min(1, 4) + min(2, 3) = 1 + 2 = 3
- (1, 3), (2, 4) -> min(1, 3) + min(2, 4) = 1 + 2 = 3
- (1, 2), (3, 4) -> min(1, 2) + min(3, 4) = 1 + 3 = 4
所以最大總和為 4
Example_2
輸入:nums = [6,2,6,5,1,2]
輸出:9
解釋:最優的分法為 (2, 1), (2, 5), (6, 6). min(2, 1) + min(2, 5) + min(6, 6) = 1 + 2 + 6 = 9
Tips
1 <= n <= 104
nums.length == 2 * n
-104 <= nums[i] <= 104
Solution
簡單的看本題就是求任意兩數的最小和
只需對序列進行簡單的排序
Code
class Solution:
def arrayPairSum(self, nums: List[int]) -> int:
nums.sort()
return sum(nums[::2])