【第十三届蓝桥杯省赛C++C组】重新排序

重新排序

给定一个数组 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⁵，1 ≤ Ai ≤ 10⁶，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。

解题思路：

先用差分求出一段区间中的每一个数被加了多少次

再用排序不等式求出最大值（贪心）

#include <iostream>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 100010;
int n, m;
int w[N], s[N];

int main()
{
    cin >> n;
    for (int i = 1; i <= n; i ++) cin >> w[i];
    cin >> m;
    while (m --)
    {
        int l, r;
        cin >> l >> r;
        s[l] ++, s[r + 1] --;  
    }
    
    // 差分数组求前缀和--->还原原数组 
    for (int i = 1; i <= n; i ++) s[i] += s[i - 1];
    
    LL sum1 = 0, sum2 = 0;
    for (int i = 1; i <= n ; i ++)
        sum1 += (LL)s[i] * w[i];
        
    sort(s + 1, s + n + 1);
    sort(w + 1, w + n + 1);
    // 两个都为升序
    for (int i = 1; i <= n; i ++)
        sum2 += (LL)s[i] * w[i];
        
    printf("%lld\n", sum2 - sum1);
    return 0;
}