给定一个数组 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。
思路分析:
本题可以贪心的思想想一下,因为数组元素是自然数递增的,所以要想求得最大的区间和,重新排列数组无非就是从小到大排序然后求各区间之和,与排序前的区间和相减就是增加了多少。
但是这样操作会忽略一个问题,如果两区间或者多区间之间有重合区间段的话,我们就要特别考虑这些重合区间,重合的点数越多我们可以让这些点分配到数组中值较大的那一部分,即我们用差分数组记录所有区间中所有点出现的次数,出现次数最多的点放到递减排序后的数组靠前的位置,这样我们就可以更充分的利用数组中较大的值,这样得出来的区间和一定是最大的。
代码如下:
import java.util.*;
class Main{
static final int N = 100010;
static int n, m;
static Integer[] a = new Integer[N], c = new Integer[N];
static long[] s = new long[N];
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
n = sc.nextInt();
for(int i = 1; i <= n; i++){
a[i] = sc.nextInt();
s[i] = s[i - 1] + a[i];
}
Arrays.fill(c, 0);
m = sc.nextInt();
long sum = 0, res = 0;
while(m-- > 0){
int l = sc.nextInt(), r = sc.nextInt();
c[l] += 1; c[r + 1] -= 1;
sum += s[r] - s[l - 1];
}
for(int i = 1; i <= n; i++) c[i] += c[i - 1];
Arrays.sort(a, 1, n + 1, (o1, o2) -> o2 - o1);
Arrays.sort(c, 1, n + 1, (o1, o2) -> o2 - o1);
for(int i = 1; i <= n; i++) res += (long)c[i] * a[i];
System.out.println(res - sum);
}
}