leetcode1589. 所有排列中的最大和（Python3、c++）

leetcode1589. 所有排列中的最大和

有一个整数数组 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 <= 10^5
• 0 <= nums[i] <= 10^5
• 1 <= requests.length <= 10^5
• requests[i].length == 2
• 0 <= starti <= endi < n

---

方法：排序+差分

思路：

本题使用排序+差分的思路。

首先，看最后的输出，是所有的查询之后的最大和，而不是我们的最大排列，因此，我们不需要模拟排列，只需要得到最大的和即可。

下面，我们考虑如何才能得到最大和，对于每次查询(x,y)，相当于对下标为[x,y]中的每个元素都查询了一次，因此我们可以遍历requests，求出对每个下标的查询次数，使用times数组，times[i]表示对下标i查询的次数。

我们如果让查询次数最多的下标的数为nums中最大值、次数第二多的下标的数为nums中的第二大的值，…，以此类推，次数最小的下标的数为nums中的最小值，即可得到我们的答案。

简单的方法是，对requests的每个(x,y)，将times[x]到times[y]每个数都+1，但是这样会超时，因为对每个(x,y)，时间复杂度为O(n)。

我们可以使用差分数组来进行优化，设s为差分数组，那么s的前缀和数组即为times，我们对times的[x,y]中每个数+1，**我们只要对s[x] += 1，s[y+1] -= 1即可。**将所有的查询遍历完之后，我们再求对s求一次前缀和，即为times数组，这样可以大大降低时间复杂度。值得注意的是，为了处理边界条件，s和times的下标都是从1开始的，下标0处两个数组的值均为0。

因此，我们的步骤如下：

• 对nums进行排序
• 遍历requests，填写差分数组s
• 根据差分数组求前缀和，得到times数组
• 将times数组排序
• nums和times两个有序数组的对应位置数据相乘，求和，即为答案。