题目介绍
有一个整数数组 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
题目分析
PART1 分析题目意思
由题目易知,nums的序列可变,request[i]=[start,end],start和end对应的是num的下标
例:<示例3>
输入:nums = [4,10,5,3,2,1], requests = [[0,2],[1,3],[1,1]]
输出:47
request[0]–>[0,2]–>nums[0]+nums[1]+nums[2]=4+10+5=19
request[1]–>[1,3]–>nums[1]+nums[2]+nums[3]=10+5+3=18
request[3]–>[1,1]–>nums[1]=10
sum = 19+18+10=47
输出的结果是最大的sum
PART2 如何输出最大结果?
例:
nums= [1,2,3,4,5,10],requests = [[0,2],[1,3],[1,1]]
nums.length()=6,
故下标范围是0-5
假设一个新的数组 count,由count记录每一个nums下标被加的次数
count[] = [1,3,2,1,0,0]
假设排好的序列为a0,a1,a2,a3,a4,a5
结果为a01+a13+a22+a31+a40+a50
要使得结果最大,结果显然易知
于是,最佳的nums序列为
[4,10,5,3,2,1] 或者 [3,10,5,4,2,1]
代码
class Solution {
public:
int maxSumRangeQuery(vector<int>& nums, vector<vector<int>>& requests) {
int len = nums.size(),i,j;
long long int result=0;
vector<int> count(len,0);
for(i=0;i<requests.size();i++){
for(j=requests[i][0];j<=requests[i][1];j++){
count[j]++;
}
}
sort(count.rbegin(),count.rend());
sort(nums.rbegin(),nums.rend());//排序
for(i=0;i<len;i++)
result+=nums[i]*count[i];
return result;
}
};
又是越界××
…
c++
python
class Solution:
def maxSumRangeQuery(self, nums: List[int], requests: List[List[int]]) -> int:
count = [0]*len(nums)
#统计每个位置出现的数的次数
for item in requests:
for i in range(item[0],item[1]+1):
count[i]+=1
count = sorted(count)
nums = sorted(nums)
sum = 0
for num,n in zip(nums,count):
sum += num*n
return sum%(pow(10,9)+7)