【Leetcode】所有排列中的最大和

第35场双周赛第二题：

有一个整数数组 nums ，和一个查询数组 requests ，其中 requests[i] = [starti, endi] 。第 i 个查询求 nums[starti] + nums[starti + 1] + ... + nums[endi - 1] + nums[endi] 的结果 ，starti 和 endi 数组索引都是 从 0 开始 的。

你可以任意排列 nums 中的数字，请你返回所有查询结果之和的最大值。

由于答案可能会很大，请你将它对 10^9 + 7 取余 后返回。

示例 1：

输入：nums = [1,2,3,4,5], requests = [[1,3],[0,1]]
输出：19
解释：一个可行的 nums 排列为 [2,1,3,4,5]，并有如下结果：
requests[0] -> nums[1] + nums[2] + nums[3] = 1 + 3 + 4 = 8
requests[1] -> nums[0] + nums[1] = 2 + 1 = 3
总和为：8 + 3 = 11。
一个总和更大的排列为 [3,5,4,2,1]，并有如下结果：
requests[0] -> nums[1] + nums[2] + nums[3] = 5 + 4 + 2 = 11
requests[1] -> nums[0] + nums[1] = 3 + 5  = 8
总和为： 11 + 8 = 19，这个方案是所有排列中查询之和最大的结果。

示例 2：

输入：nums = [1,2,3,4,5,6], requests = [[0,1]]
输出：11
解释：一个总和最大的排列为 [6,5,4,3,2,1] ，查询和为 [11]。

示例 3：

输入：nums = [1,2,3,4,5,10], requests = [[0,2],[1,3],[1,1]]
输出：47
解释：一个和最大的排列为 [4,10,5,3,2,1] ，查询结果分别为 [19,18,10]。

提示：

• n == nums.length
• 1 <= n <= 105
• 0 <= nums[i] <= 105
• 1 <= requests.length <= 105
• requests[i].length == 2
• 0 <= starti <= endi < n

Solution:

首先最容易想到暴力法，但是超时了。

public static int maxSumRangeQuery(int[] nums, int[][] requests) {
    Arrays.sort(nums);
    int[] map = new int[nums.length];
    for (int[] pair : requests) {
        int i = pair[0];
        int j = pair[1];
        while (i <= j) {
            map[i++]++;
        }
    }
    Arrays.sort(map);
    int sum = 0;
    for (int i = map.length - 1; i >= 0; i--) {
        if (map[i] == 0 || nums[i] == 0)
            return sum;
        sum = sum + nums[i] * map[i];
        sum = (int) (sum % (1e9 + 7));
    }
    return sum;
}

更好的解法是差分：

public static int maxSumRangeQuery(int[] nums, int[][] requests) {
    int n = nums.length;
    int[] c = new int[n];
    // 构建差分数组 c 的过程
    for (int[] pair : requests) {
        int i = pair[0], j = pair[1];
        c[i]++;
        if (j + 1 < n)
            c[j + 1]--;
    }
    // 从差分数组反推目标数组
    for (int i = 1; i < n; i++) {
        c[i] += c[i - 1];
    }
    // 对两个数组升序排序
    Arrays.sort(nums);
    Arrays.sort(c);
    int sum = 0;
    // 由于升序排序，最大值在最后，所以从后往前遍历，可以在任意一个数等于0时快速跳出循环
    for (int i = n - 1; i >= 0; i--) {
        if (c[i] == 0 || nums[i] == 0)
            return sum;
        sum = sum + nums[i] * c[i];
        sum = (int) (sum % (1e9 + 7));
    }
    return sum;
}