给定一个数组 AAA 和一些查询 Li,RiL_i,R_iLi,Ri,求数组中第 LiL_iLi 至第 RiR_iRi 个元素之和。
小蓝觉得这个问题很无聊,于是他想重新排列一下数组,使得最终每个查询结果的和尽可能地大。
小蓝想知道相比原数组,所有查询结果的总和最多可以增加多少?
输入格式
输入第一行包含一个整数 nnn。
第二行包含 nnn 个整数 A1,A2,⋅⋅⋅,AnA_1,A_2,···,A_nA1,A2,⋅⋅⋅,An,相邻两个整数之间用一个空格分隔。
第三行包含一个整数 mmm 表示查询的数目。
接下来 mmm 行,每行包含两个整数 Li、RiL_i、R_iLi、Ri,相邻两个整数之间用一个空格分隔。
输出格式
输出一行包含一个整数表示答案。
数据范围
对于 30%30\%30% 的评测用例,n,m≤50n,m≤50n,m≤50;
对于 50%50\%50% 的评测用例,n,m≤500n,m≤500n,m≤500;
对于 70%70\%70% 的评测用例,n,m≤5000n,m≤5000n,m≤5000;
对于所有评测用例,1≤n,m≤105,1≤Ai≤106,1≤Li≤Ri≤n1≤n,m≤10^5,1≤A_i≤10^6,1≤L_i≤R_i≤n1≤n,m≤105,1≤Ai≤106,1≤Li≤Ri≤n。
输入样例:
5
1 2 3 4 5
2
1 3
2 5
输出样例:
4
样例解释
原来的和为 6+14=206+14=206+14=20,重新排列为 (1,4,5,2,3)(1,4,5,2,3)(1,4,5,2,3) 后和为 10+14=2410+14=2410+14=24,增加了 444。
#include<iostream>
#include<algorithm>
#include<vector>
#define x first
#define y second
using namespace std;
const int N = 100010;
typedef pair<int, int> PII;
typedef long long LL;
int n, m;
int a[N], c[N];
LL b[N], d[N];
int l[N], r[N];
int main(){
scanf("%d", &n);
for(int i = 1; i <= n; i++) scanf("%d", &a[i]), b[i] = a[i] + b[i-1];
scanf("%d", &m);
LL s1 = 0;
for(int i = 0; i < m; i++) {
scanf("%d%d", &l[i], &r[i]);
s1 += b[r[i]] - b[l[i] - 1];
c[l[i]] ++, c[r[i] + 1]--;
}
vector<PII> v;
for(int i = 1; i <= n; i++) c[i] += c[i-1], v.push_back({c[i], i});
sort(v.begin(), v.end(), greater<PII>());
sort(a + 1, a + n + 1, greater<int>());
for(int i = 0; i < n; i++) d[v[i].y] = a[i + 1];
for(int i = 1; i <= n; i++) d[i] += d[i-1];
LL s2 = 0;
for(int i = 0; i < m; i++) s2 += d[r[i]] - d[l[i] - 1];
printf("%lld", s2 - s1);
return 0;
}