4655. 重新排序

4655. 重新排序

原题

题目链接

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

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

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

输入格式

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

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

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

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

输出格式

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

数据范围

对于 30%30% 的评测用例，n,m≤50,≤50；
对于 50%50% 的评测用例，n,m≤500,≤500；
对于 70%70% 的评测用例，n,m≤5000,≤5000；
对于所有评测用例，1≤n,m≤1051≤,≤105，1≤Ai≤1061≤≤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。

题解

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=100010;
bool cmp(int a,int b){
	return a>b;
}
int main()
{
	int n,m,s[N]={0},a[N];
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	scanf("%d",&m);
	int l,r;
	for(int i=0;i<m;i++)
	{
		scanf("%d %d",&l,&r);
		s[l]++;s[r+1]--;
	}
	
//	差分统计次数
	for(int i=1;i<=n;i++) s[i]+=s[i-1];

//	求出原始和以及排序和 
	long long psum=0,nsum=0;
	for(int i=1;i<=n;i++) psum+=(long long)s[i]*a[i];
	sort(a+1,a+n+1,cmp);
	sort(s+1,s+1+n,cmp);
	for(int i=1;i<=n;i++) nsum+=(long long)s[i]*a[i];
	 
	printf("%lld",nsum-psum);
	return 0;
}

• 最开始想的是可以用一个数组来统计各个地方的数出现的次数，但是时间复杂度可能可以达到O（n2），遍历m嵌套遍历赋值

• 看了闫总的思路笔记：1.看到说给数组，画条横线，2、前缀和算法（前缀和的逆运算-差分），使用差分来快速统计每个位置被加了多少次，O（n）的复杂度，下面就好做多了

• 前面统计的时间复杂度可以了的话，对A和对统计数组排序nlogn，+106+106+

• 一个值得注意的点：psum+=(long long)s[i]*a[i];，int和int乘完估计不会自动转成long，乘的结果仍保存为int类型，从而导致出现错误，（我在调试的时候始终有一个测试点过不去，而且是数据大小比正确答案差了两个数量级的那种，然后最后对比闫总的代码发现少了个这）