1. 程式人生 > 其它 >力扣 leetcode 561. 陣列拆分 I (python)

力扣 leetcode 561. 陣列拆分 I (python)

技術標籤: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. (1, 4), (2, 3) -> min(1, 4) + min(2, 3) = 1 + 2 = 3
  2. (1, 3), (2, 4) -> min(1, 3) + min(2, 4) = 1 + 2 = 3
  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])

Result

在這裡插入圖片描述