给定一个数组 A 和一些查询 Li,Ri,求数组中第 Li 至第 Ri 个元素之和。
小蓝觉得这个问题很无聊,于是他想重新排列一下数组,使得最终每个查询结果的和尽可能地大。
小蓝想知道相比原数组,所有查询结果的总和最多可以增加多少?
输入格式
输入第一行包含一个整数 n。
第二行包含 n 个整数 A1,A2,···,An,相邻两个整数之间用一个空格分隔。
第三行包含一个整数 m 表示查询的数目。
接下来 m 行,每行包含两个整数 Li、Ri,相邻两个整数之间用一个空格分隔。
输出格式
输出一行包含一个整数表示答案。
数据范围
对于 30% 的评测用例,n,m≤50;
对于 50% 的评测用例,n,m≤500;
对于 70% 的评测用例,n,m≤5000;
对于所有评测用例,1≤n,m≤105,1≤Ai≤106,1≤Li≤Ri≤n。
输入样例:
5
1 2 3 4 5
2
1 3
2 5
输出样例:
4
样例解释
原来的和为 6+14=20,重新排列为 (1,4,5,2,3) 后和为 10+14=24,增加了 4。
这道题我们的思路就是统计出每一次询问之后,对应位置的元素需要统计多少次。
比方说我们上面的的样例中的数据为:1 2 3 4 5
然后我们的两次询问分别为1 3
也就是说1-3号元素分别会被统计一次
0 | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
0 | 1 | 1 | 1 |
然后询问2 5,也就是说2-5号元素又会被统计一次
0 | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
0 | 1 | 2 | 2 | 1 | 1 |
然后我们对我们出现次数的数组进行排序,得到
0 | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
0 | 1 | 1 | 1 | 2 | 2 |
然后我们也将我们的原数组进行排序,得到
0 | 1 | 2 | 3 | 4 | 5 |
---|---|---|---|---|---|
0 | 1 | 2 | 3 | 4 | 5 |
然后我们只要将两个数组对应的位置相乘就能够得到我们的结果
1*1+1*2+1*3+2*4+2*5=24
所以我们的核心思路就是:统计我们每一个位置的元素需要被计算几次,然后将我们数组中较大的数排到这几个出现次数更多的位置上去。
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main()
{
int n,m;
//用来统计初始的数据和
long long sum1=0;
//用来统计优化之后的数据和
long long sum2=0;
cin>>n;
//用来存放我们初始的数据
vector<long long> number(n+10,0);
//用来存放我们的数据出现次数
vector<long long> quest(n+10,0);
for(int i=1;i<=n;i++)
{
int tmp;
cin>>tmp;
number[i]=tmp;
}
cin>>m;
while(m--)
{
int left,right;
cin>>left>>right;
//使用差分的方法,也就是每一个位置存的是后一项减前一项的值
//所以如果我们想要将[left,right]之间的数据都加上1的话,
//我们就需要将left位置++
//在我们的right+1的位置--
quest[left]++;
quest[right+1]--;
}
//将我们的差分的数据进行还原,得到每一个位置真正需要出现多少次
for(int i=1;i<=n;++i)
{
quest[i]+=quest[i-1];
}
//统计我们原始的累加和
for(int i=1;i<=n;++i)
{
sum1+=number[i]*quest[i];
}
//排序
sort(number.begin()+1,number.begin()+n+1);
sort(quest.begin()+1,quest.begin()+n+1);
//统计我们优化之后的累加和
for(int i=1;i<=n;++i)
{
sum2+=number[i]*quest[i];
}
cout<<sum2-sum1<<endl;
}