1971: 安排座位
Time Limit: 2 Sec Memory Limit: 128 Mb
Description
一年一度的暑期集训又开始了!
作为老人的小明非常忧伤,因为他要给所有的新人安排座位。由于安排给新人的座位上的机器可能有各种毛病(比如很卡,上不了网之类的),这些问题的出现都会让新人的训练热情下降。为了让更多的新人能够留下,小明自然希望大家的热情都是高涨的。
对于每个新人,都会有一个热情值ai,而每个座位都会有一个热情耗损值bi,如果第i个新人坐在第j个位置,那这位同学对整个集训队热情值的贡献就是(ai - bj) ^2。现在给出所有新人的热情值,所有位置的热情耗损值,你能告诉小明采用最合理的位置安排方式后,能得到的最大的集训队热情值是多少?
当然,每个位置只能坐一个新人,每个新人也必须坐在某个位置上
Input
第一行一个数字T表示数据组数
每组数据包括三行:
第一行为一个整数n,表示新人的人数
第二行为n个整数,第i个数字表示第i个同学的热情值ai
第三行为n个整数,第i个数字表示第i个座位的热情耗损值为bi
其中T<=10 , 0<=ai , bi <=100, 1<=n<=100000
Output
输出一行只包含一个整数,表示集训队热情值的最大值
Sample Input
2
3
2 5 1
0 0 1
3
2 5 1
3 2 5
Sample Output
29
26
Hint
Source
2017年7月月赛
题目链接:http://acm.csu.edu.cn/csuoj/problemset/problem?pid=1971
题解:本题只需要对n个热情值和n个热情损耗值进行升序和降序的排列即可迅速得出答案,可以利用multiset容器的特点。
#include<iostream>
#include<set>
using namespace std;
int main(){
int T;
scanf("%d",&T);
int n;
multiset<int>m1;
multiset<int>m2;
int tmp,tmp2;
while(T--){
m1.clear(); //注意一定要清空容器
m2.clear();
scanf("%d",&n);
int t=n;
while(t--){
scanf("%d",&tmp);
m1.insert(tmp);
}
t=n;
while(t--){
scanf("%d",&tmp2);
m2.insert(tmp2);
}
int sum=0;
multiset<int>::iterator it;
multiset<int>::iterator it2;
it=m1.begin();
it2=m2.end();
it2--;
t=n;
for(int i=0;i<n;i++){
sum+=((*it)-(*it2))*((*it)-(*it2));
it++;
it2--;
}
printf("%d\n",sum);
}
return 0;
}