leetcode 之 561. 数组拆分 I

223 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第9天,点击查看活动详情

一 描述

561. 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)

给定长度为 2n 的整数数组 nums ,你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), ..., (an, bn) ,使得从 1 到 n 的 min(ai, bi) 总和最大。

返回该 最大总和 。

示例 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

示例 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

提示:

  • 1 <= n <= 10^4
  • nums.length == 2 * n
  • -10^4 <= nums[i] <= 10^4

二 分析

本质思路是将整个队列按从小到大的方式排序,然后从两个相近的数中选取出最小的,取和。使用冒泡法会超出计算时间,因为选用额外增加数组,将大小作为两个数组的下角标,从而进行排序。

三 答案

class Solution {
public:
    int arrayPairSum(vector<int>& nums) {
       /*
        int arr[nums.size()]={0};
        int temp=0;
        for(int i=0;i<nums.size();i++)
        arr[i]=nums[i];
        for(int i=0;i<nums.size()-1;i++)
        {
            for(int j=i+1;j<nums.size();j++)
            if(arr[i]>arr[j]) {temp=arr[i];arr[i]=arr[j];arr[j]=temp;}   //冒泡排序会超出时间限制
        }
        int res=0;
        for(int i=0;i<nums.size();i+=2){res+=arr[i];}
        return res;*/
        int n[20001]={0},i=0,j=0,sum=0;
        for(i=0;i<nums.size();i++)
        n[nums[i]+10000]++;        //变相将nums按顺序大小排好,并将顺序存储在n[i]中
        for(i=0;i<20001;)
        {
            if(n[i])
            {
                if(j%2==0) sum+=i-10000;     //按照n[i]中存储的顺序,求得nums[i]
                j++;                  //只有n[i]中有数时,即nums存在这个数时,j才增加,将j变为下角标
                n[i]--;           //此步防止nums中含有两个相同的数
            }
            else i++;
        }
        return sum;
    }
};

参考

从示例入手分析解法,三种代码练习 Python 语言 - 数组拆分 - 力扣(LeetCode)