1589. 所有排列中的最大和

leetcode 1589. 所有排列中的最大和

题目链接

问题描述

• 题目描述：
    有一个整数数组 nums ，和一个查询数组 requests ，其中 requests[i] = [starti, endi] 。第 i 个查询求 nums[$star{t}_i$] + nums[$star{t}_i+1$] + ... + nums[$en{d}_i-1$] + nums[$en{d}_i$] 的结果 ，$star{t}_i$ 和 $en{d}_i$ 数组索引都是 从 0 开始 的。
  你可以任意排列 nums 中的数字，请你返回所有查询结果之和的最大值。
  由于答案可能会很大，请你将它对 $10^9+7$ 取余 后返回
• 参数要求：
  $n==nums.length$
  $1<=n<=10^5$
  $0<=nums[i]<=10^5$
  $1<=requests.length<=10^5$
  $requests[i].length==2$
  $0<=star{t}_i<=en{d}_i<n$

思路

1. 首先需要求出的一种排列可能（实际上可能存在多种排列）需要满足以下条件，即按照对应区间求和最大
2. 在分析区间中我们不难发现，对于requests中重复最多的下标降序排列，得到一个数组count，如果要使得最后的结果最大，那就要用最大的count搭配nums中未被使用最大的元素。
3. 因此需要对nums递增排序，并且排序并不影响结果。
4. 这里需要注意的是，统计count的方法，如果是直接统计，也就是以遍历的方式统计，那么最后会出现超时。例如：对于request[i]的统计

 			int start = requests[i][0];
            int end = requests[i][1];
            while (start<=end){
                Integer count = map.getOrDefault(start,0);//如果没有就返回0
                map.put(start,count+1);
                start++;
            }

5. 因此采用差分数组统计区间次数，最后通过一次前缀数组就可以获得每一个下标的包含的次数。类似题目1109. 航班预订统计。
   
   

代码

public int maxSumRangeQuery(int[] nums, int[][] requests) {
        int len = nums.length;
        Arrays.sort(nums);
        int[] count = new int[len];
        for (int i = 0; i < requests.length; i++) {
            int start = requests[i][0];
            int end = requests[i][1];
            count[start]++;
            if (end<len-1){
                count[end+1]--;
            }
        }
        for (int i = 1; i < len; i++) {
            count[i] += count[i-1];
        }
        Arrays.sort(count);
        long res = 0;
        int mod = (int)1e9+7;
        int indexCount = len-1;
        int indexNum = len-1;
        while (indexCount>=0&&count[indexCount]!=0){
            res = (res+(long)nums[indexNum]*count[indexCount])%mod;
            indexCount--;
            indexNum--;
        }
        return (int)res;
    }