数据结构习题--数组拆分

数据结构习题–数组拆分

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

方法一：直接遍历

分析

我们分析其实，要使每个组合的最小值相加的和最大，那么问题在于每个组合的最小值，我们通过对数组进行排序，取每两个数的前一个数，进行累加和，最后得到总和

代码

import java.util.Arrays;

public class demergeArray {
    public int arrayPairSum(int[] nums) {
        Arrays.sort(nums);
        int sum = 0;
        for (int i = 0; i < nums.length; i += 2) {
            sum += nums[i];
        }
        return sum;
    }
}

复杂度

时间复杂度：O(nlogn)，即为对数组 nums 进行排序的时间复杂度。
空间复杂度：O(logn)，即为排序需要使用的栈空间

方法二：计数排序

分析

先统计每个数出现的次数，把每个数的值转换为一个新数组的索引，然后这个数每出现一次，对应索引的数就增加1，比如123234，那么新的数组的索引为1的值就加1，然后索引为2的值也加1，以此类推…
对于这个题，我们只需要把新数组每个元素的值除以2(得到个数)然后乘以索引(原来数的大小),但是对于个数，可能为奇数，或者偶数，偶数无须特殊处理，但是对于奇数，需要往后一个数借1，当然后面一个数的个数也需要更新

代码

class Solution {
        public int arrayPairSum(int[] nums) {
            int[] count = new int[20001];
            for(int x:nums){
                count[x + 10000]++;
            }
            // borrow是向上借一个
            int ret = 0, borrow=0;
            for(int i=-10000;i<=10000;i++){
                // count[i]是总共出现了多少个， -borrow是i-1有没有借走一个
                // +1是为了保证，count[i] - borrow是奇数的情况下，多算一个
                ret += (count[i+10000] - borrow + 1) / 2 * i;

                // 更新borrow，如果count[i] - borrow是奇数，就得从i+1借一个
                // +2是java语言防止-1 % 2 == -1,因为每个计数的位置可能为0，而borrow可能为0,1，为了满足取模，所以加2
                borrow = (count[i+10000] - borrow + 2) % 2;
            }
            return ret;
        }
    }