问题描述
给定一个数组 A 和一些查询Li,Ri, 求数组中第Li 至第 Ri 个元素之和。
小蓝觉得这个问题很无聊, 于是他想重新排列一下数组, 使得最终每个查 询结果的和尽可能地大。小蓝想知道相比原数组, 所有查询结果的总和最多可 以增加多少?
输入格式
输入第一行包含一个整数 n 。
第二行包含 n 个整数 A1,A2,⋯,An, 相邻两个整数之间用一个空格分隔。
第三行包含一个整数 m 表示查询的数目。
接下来 m 行, 每行包含两个整数 Li、Ri, 相邻两个整数之间用一个空格分 隔。
输出格式
输出一行包含一个整数表示答案。
样例输入
5
1 2 3 4 5
2
1 3
2 5
样例输出
4
样例说明
原来的和为6+14=20, 重新排列为 (1,4,5,2,3) 后和为 10+14=24, 增 加了 4。
评测用例规模与约定
对于 30%30% 的评测用例,n,m≤50;
对于 50%50% 的评测用例,n,m≤500;
对于 70%70% 的评测用例, n,m≤5000;
对于所有评测用例, 1≤n,m≤105,1≤Ai≤106,1≤Li≤Ri≤106 。
运行限制
- 最大运行时间:1s
- 最大运行内存: 512M
此题无需前期的差分处理,直接进行关系操作,最后进行前缀和即可。
#include<bits/stdc++.h>
using namespace std;
long long int data[100005];
long long int frequency[100005];//计算位置频率
int main()
{
long long int n,m,x,sum=0,num=0,a,b;
cin>>n;
for(int i=0;i<n;i++){
cin>>data[i];
}
cin>>m;
for(int i=0;i<m;i++){
cin>>a>>b;//输入位置
a=a-1;
b=b-1;
frequency[a]++;//差分关系操作
frequency[b+1]--;
}
for(int i=1;i<n;i++){
frequency[i]=frequency[i]+frequency[i-1];//差分前缀和
}
for(int i=0;i<n;i++){
sum=sum+frequency[i]*data[i];
}
sort(data,data+n,greater<long long int>());
sort(frequency,frequency+n,greater<long long int>());
for(int i=0;i<n;i++){
num+=frequency[i]*data[i];
}
cout<<num-sum;
return 0;
}