蓝桥杯 2022 重新排序

问题描述

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

小蓝觉得这个问题很无聊,于是他想重新排列一下数组,使得最终每个查询结果的和尽可能地大。小蓝想知道相比原数组,所有查询结
果的总和最多可 以增加多少?

输入格式

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

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

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

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

输出格式

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

样例输入

5

1 2 3 4 5

2

1 3

2 5

样例输出

4

样例说明

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

评测用例规模与约定

对于30%的评测用例,n,m≤50;

对于50%的评测用例,n,m≤500;

对于70%的评测用例,n,m≤5000;

对于所有评测用例,1≤n,m≤105,1≤Ai≤106,1≤Li≤Ri≤106。

思路：

        这个题多次用到前缀和以及差分思想；

        

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N = 1e6+10;
int n,m,t,d;
int a[N],ct[N],s[N];
string sa,sb;
vector<int>vt;
void countTime(int l,int r,int x){  //这个函数的作用是标记每次从l开始到r会用到的次数
	ct[l]+=x;  //因为后面会再用一次统计次数的前缀和，所以这里使左边界加1次即可，
	ct[r+1]-=x; //右边界+1的地方减一拦截范围。
}
signed main(){
	int res1=0,res2=0;
	int l,r;
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>a[i];  //原数组
		s[i] = s[i-1]+a[i];  //原数组前缀和数组
	}
	cin>>m;
	while(m--){
		cin>>l>>r; 
		res1 += s[r]-s[l-1]; //求原数和
		countTime(l,r,1);  //标记l->r遍历次数
	}
	for(int i=1;i<=n;i++)ct[i]+=ct[i-1];//又一次前缀和统计出现次数
	sort(a+1,a+n+1); //原数组从小到大排序
	sort(ct+1,ct+n+1);  //统计次数数组从小到大排序
	int index = n; 
	while(index){ //因为没统计的说明不会用到也就是0，不会加到结果数组中
		res2+=ct[index]*a[index];  //这里直接用次数*原数组数值再相加；
		index--;
	} 
	cout<<res2-res1<<endl;
//	cout<<res1<<endl;
    return 0;
}