给定一个数组 A A A 和一些查询 L i , R i L_i,R_i Li,Ri,求数组中第 L i L_i Li 至第 R i R_i Ri 个元素之和。
小蓝觉得这个问题很无聊,于是他想重新排列一下数组,使得最终每个查询结果的和尽可能地大。
小蓝想知道相比原数组,所有查询结果的总和最多可以增加多少?
输入格式
输入第一行包含一个整数 n n n。
第二行包含 n n n 个整数 A 1 , A 2 , ⋅ ⋅ ⋅ , A n A_1,A_2,⋅⋅⋅,A_n A1,A2,⋅⋅⋅,An,相邻两个整数之间用一个空格分隔。
第三行包含一个整数 m m m 表示查询的数目。
接下来 m m m 行,每行包含两个整数 L i 、 R i L_i、R_i Li、Ri,相邻两个整数之间用一个空格分隔。
输出格式
输出一行包含一个整数表示答案。
数据范围
对于 30 30 30% 的评测用例, n , m ≤ 50 n,m≤50 n,m≤50;
对于 50 50 50% 的评测用例, n , m ≤ 500 n,m≤500 n,m≤500;
对于 70 70 70% 的评测用例, n , m ≤ 5000 n,m≤5000 n,m≤5000;
对于所有评测用例, 1 ≤ n , m ≤ 105 , 1 ≤ A i ≤ 106 , 1 ≤ L i ≤ R i ≤ n 1≤n,m≤105,1≤A_i≤106,1≤L_i≤R_i≤n 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。
很容易想到能够用差分的方法来求出所有操作覆盖到每个点的次数,如果使用c[]
存储那么原本能够得到的总和就是
c 1 ∗ a 1 + c 2 ∗ a 2 + . . . . . . + c i ∗ a i c_1*a_1 + c_2*a_2 + ...... + c_i*a_i c1∗a1+c2∗a2+......+ci∗ai
可以想到如果把所有的数值更大的 a a a放到覆盖次数更多的点上就可以得到最大的结果,这一步不要想怎么去把 a a a放到相应的位置,而是直接可以对两个数组进行同样的递增或递减顺序排序,排序之后,最大的 a a a元素必然对齐了最大的 c c c元素。
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 1e5+10;
#define ll long longint n,m;
int b[N];
int a[N];
ll c[N];void insert(int l,int r,int c){b[l] += c;b[r+1] -= c;
}int main(){cin >> n;for(int i = 1;i <= n;i++)cin >> a[i];cin >> m;while(m--){int l,r;cin >> l >> r;insert(l,r,1);}for(int i = 1;i <= n;i++)c[i] = c[i-1] + b[i];//求出操作覆盖每个点的次数ll oriRes = 0;for(int i = 1;i <= n;i++)oriRes += c[i]*a[i];//求出原本的答案sort(c+1,c+1+n);sort(a+1,a+1+n);ll res = 0;for(int i = 1;i <= n;i++)res += c[i]*a[i];cout << res - oriRes;return 0;
}