LeetCode刷题(175)~所有排列中的最大和【第35场双周赛:题目二 扫描线】

阿里云国内75折 回扣 微信号:monov8
阿里云国际,腾讯云国际,低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号:monov8 飞机:@monov6


题目描述

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

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

由于答案可能会很大,请你将它对 109 + 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

解答 By 海轰

提交代码(超时)

int maxSumRangeQuery(vector<int>& nums, vector<vector<int>>& requests) {
int num=nums.size();
vector<int> a(num,0);
for(int i=0;i<requests.size();++i)
{
for(int j=requests[i][0];j<=requests[i][1];++j)
++a[j];
}
sort(a.begin(),a.end());
sort(nums.begin(),nums.end());
long long ans=0;
for(int i=0;i<num;++i)
{
if(a[i]!=0)
{
ans+= (a[i]*nums[i]);
}
}
return ans%1000000007;
}

运行结果

LeetCode刷题(175)~所有排列中的最大和【第35场双周赛:题目二  扫描线】_提交代码

提交代码(优化 扫描线)

int maxSumRangeQuery(vector<int>& nums, vector<vector<int>>& requests) {
int num=nums.size();
vector<int> a(num+1,0);// 注意长度需要+1 因为后面有requests[i][1]+1 避免溢出
for(int i=0;i<requests.size();++i)
{
++a[requests[i][0]];
--a[requests[i][1]+1];
}
for(int i=1;i<num+1;++i)
a[i]+=a[i-1];
sort(a.begin(),a.begin()+num);// 这里注意不是a.begin()-a.end() 因为最后一个数不需要进行统计
sort(nums.begin(),nums.end());
long long ans=0;
for(int i=0;i<num;++i)
{
if(a[i]!=0)
{
ans+= (a[i]*nums[i]);
}
}
return ans%1000000007;
}

运行结果

LeetCode刷题(175)~所有排列中的最大和【第35场双周赛:题目二  扫描线】_扫描线_02

题目来源

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-sum-obtained-of-any-permutation


阿里云国内75折 回扣 微信号:monov8
阿里云国际,腾讯云国际,低至75折。AWS 93折 免费开户实名账号 代冲值 优惠多多 微信号:monov8 飞机:@monov6

“LeetCode刷题(175)~所有排列中的最大和【第35场双周赛:题目二 扫描线】” 的相关文章