leetcode_5505. 所有排列中的最大和
目录
一、题目内容
有一个整数数组 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
二、解题思路
1.记录requests中范围内数字的使用次数，存入diff中。
2.注意存储次数的方式不能暴力的双重for，否则导致超时。
3.利用射线传递和截断的思想，每次都将自己是否作为起点或终点的数量与前一个数字在requests中各个范围中包含个数相加，得到自己的使用次数。
三、代码
-
class Solution:
-
def maxSumRangeQuery(self, nums: list, requests: list) -> int:
-
n = len(nums)
-
diff = [0] * n
-
for s, e in requests:
-
diff[s] += 1
-
if e + 1 < n:
-
diff[e + 1] -= 1
-
for i in range(1, n):
-
diff[i] += diff[i - 1]
-
nums.sort()
-
diff.sort()
-
ans = 0
-
for i in range(n):
-
ans += nums[i] * diff[i]
-
return ans % (10 ** 9 + 7)
-
-
-
if __name__ == '__main__':
-
nums = [1, 2, 3, 4, 5]
-
requests = [[1, 3], [0, 1]]
-
-
s = Solution()
-
ans = s.maxSumRangeQuery(nums, requests)
-
print(ans)
文章来源: nickhuang1996.blog.csdn.net，作者：悲恋花丶无心之人，版权归原作者所有，如需转载，请联系作者。
原文链接：nickhuang1996.blog.csdn.net/article/details/108692221
- 点赞
- 收藏
- 关注作者
评论（0）