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)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 29.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值