题目描述
给定一个数组 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% 的评测用例,n, m ≤ 50 ;
对于 50% 的评测用例,n, m ≤ 500 ;
对于 70% 的评测用例,n, m ≤ 5000 ;
对于所有评测用例,1 ≤ n, m ≤ 10^5,1 ≤ Ai≤ 10^6,1 ≤ Li ≤ Ri≤ 10^6 。
参考代码:
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<vector>
using namespace std;
int main()
{
int n;
cin>>n;
vector<int> a(n);
for(int i=0;i<n;i++)
{
cin>>a[i];
}
int m;
int total=0;
cin>>m;
int* b=new int[m];
int* c=new int[m];
vector<int> d(n);//每一项出现次数
for(int i=0;i<m;i++)
{
cin>>b[i]>>c[i];
for(int j=b[i]-1;j<=c[i]-1;j++)
{
total+=a[j];
d[j]++;
}
}
int sum=0;
vector<int> f=d;
sort(f.begin(),f.end());
for(int i=0;i<n;i++)
{
sum+=a[i]*f[i];
}
#if 0
for(int i=0;i<n;i++)//测试
{
cout<<d[i]<<' ';
}
cout<<"未排序时"<<endl;
for(int i=0;i<n;i++)
{
cout<<f[i]<<' ';
}
cout<<"排序时"<<endl;
cout<<total<<endl;
cout<<sum<<endl;
#endif
cout<<sum-total;
return 0;
}
运行结果:(打开测试时)
5
1 2 3 4 5
2
1 3
2 5
1 2 2 1 1 未排序时
1 1 1 2 2 排序时
20
24
4