Acwing 4655.重新排序 差分 权重

给定一个数组 A 和一些查询 Li,Ri求数组中第 Li 至第 Ri 个元素之和。

小蓝觉得这个问题很无聊，于是他想重新排列一下数组，使得最终每个查询结果的和尽可能地大。

小蓝想知道相比原数组，所有查询结果的总和最多可以增加多少?

输入格式

输入第一行包含一个整数 n

第二行包含 n 个整数 A1,A2,···,An，相邻两个整数之间用一个空格分隔。

第三行包含一个整数 m 表示查询的数目。

接下来 m行，每行包含两个整数 Li、Ri相邻两个整数之间用一个空格分隔。

输出格式

输出一行包含一个整数表示答案。

数据范围

对于 30%30% 的评测用例，n,m≤50
对于 50%50% 的评测用例，n,m≤500；
对于 70%70% 的评测用例，n,m≤5000
对于所有评测用例，1≤n,m≤10^5，1≤Ai≤106，1≤Li≤Ri≤n1

输入样例：

5
1 2 3 4 5
2
1 3
2 5

输出样例：

4

样例解释

原来的和为 6+14=206+14=20，重新排列为 (1,4,5,2,3)(1,4,5,2,3) 后和为 10+14=2410+14=24，增加了 44。

思路：

首先我们容易的看出，这个题是一道差分的题。因为给出的操作区间，所以我们可以维护一个times数组记录每个位置被操作的次数。因为此题是求解最大增加了多少，因此我们就需要给出现次数最多的地方放值更加大的数字。所以我们将次数数组和原数组进行排序，然后根据权值再相乘相加。

代码：

import java.util.*;
public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int len = sc.nextInt();
        int[] arr = new int[len];
        for (int i = 0; i < len; i++) {
            arr[i] = sc.nextInt();
        }
        int m = sc.nextInt();
        int[] times = new int[len+2];
        for (int i = 0; i < m; i++) {
            int l = sc.nextInt();
            int r = sc.nextInt();
            times[l]++;
            times[r+1]--;
        }
        long pre = 0;
        for (int i = 1; i < times.length; i++) {
            times[i] += times[i-1];
        }
        for (int i = 0; i < len; i++) {
            pre += (long) arr[i]*times[i+1];
        }
        Arrays.sort(times,1,times.length-1);
        Arrays.sort(arr);
        long now = 0;
        for (int i = 0; i < len; i++) {
            now += (long)arr[i]*times[i+1];
        }
        System.out.println(now- pre);
    }
}