【C++算法】重新排序

题目描述

给定一个数组 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 ≤ 10^5，1 ≤ Ai≤ 10^6，1 ≤ Li ≤ Ri≤ 10^6 。

参考代码：

#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<vector>
using namespace std;
int main()
{
	int n;
	cin>>n;
	vector<int> a(n);
	for(int i=0;i<n;i++)
	{
		cin>>a[i];
	}
	int m;
	int total=0;
	cin>>m;
	int* b=new int[m];
	int* c=new int[m];
	vector<int> d(n);//每一项出现次数
	for(int i=0;i<m;i++)
	{
		cin>>b[i]>>c[i];
		for(int j=b[i]-1;j<=c[i]-1;j++)
		{
			total+=a[j];
			d[j]++;
		}
	}

	int sum=0;
	vector<int> f=d;
	sort(f.begin(),f.end());
	for(int i=0;i<n;i++)
	{
		sum+=a[i]*f[i];
	}
	
#if 0
	for(int i=0;i<n;i++)//测试
	{
		cout<<d[i]<<' ';
	}
	cout<<"未排序时"<<endl;
	for(int i=0;i<n;i++)
	{
		cout<<f[i]<<' ';
	}
	cout<<"排序时"<<endl;
	cout<<total<<endl;
	cout<<sum<<endl;
#endif
	cout<<sum-total;

	return 0;
}

 运行结果：（打开测试时）

5

1 2 3 4 5
2
1 3
2 5
1 2 2 1 1 未排序时
1 1 1 2 2 排序时
20
24
4