LeetCode:561. Array Partition(數組分割)

給定長度爲 2n 的數組, 你的任務是將這些數分成 n 對, 例如 (a1, b1), (a2, b2), ..., (an, bn) ,使得從1 到 n 的 min(ai, bi) 總和最大。

示例 1:

輸入: [1,4,3,2]

輸出: 4
解釋: n 等於 2, 最大總和爲 4 = min(1, 2) + min(3, 4).
提示:n 是正整數,範圍在 [1, 10000].數組中的元素範圍在 [-10000, 10000].

 


著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。


java實現方法1:

   /**
     * 計算數組中一部分的和
     *
     * @param nums 數組
     * @return 計算和
     */
    public int arrayPairSum(int[] nums) {
        int sum = 0;
        Arrays.sort(nums);
        for (int i = 0; i < nums.length - 1; i += 2) {
            int min = Math.min(nums[i], nums[i + 1]);
            sum += min;
        }
        return sum;
    }

時間複雜度:O(n.logn)

空間複雜度:O(n)


python實現方法1:

'''
分組最小求和
author:zhangyu
date:2020.1.15
'''
from typing import List


def array_pair_sum(nums: List[int]) -> int:
    '''
        對數組求和,分組最小
    Args:
        nums: 數組
    Returns:
        求和
    '''
    sum = 0
    nums.sort()
    i = 0
    while i < len(nums) - 1:
        min_num = min(nums[i], nums[i + 1])
        sum += min_num
        i += 2
    return sum


if __name__ == '__main__':
    nums = [1, 2, 3, 4]
    sum = array_pair_sum(nums)
    print(sum)

時間複雜度:O(n.logn)

空間複雜度:O(n)


源碼github地址:https://github.com/zhangyu345293721/leetcode

發佈了240 篇原創文章 · 獲贊 64 · 訪問量 9萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章