Honk's pool ( STL multiset )
multiset 和 set 唯一的不同是 set 中一个值只能出现一次,而multiset中一个值可以出现多次。
题目链接:https://nanti.jisuanke.com/t/41406
题意: n个数,每次挑一个最大的减一,挑一个最小的加一, 问k次操作后最大值和最小值差多少。
思路:纯模拟。学习一下multiset的应用。好像一次sort后直接for循环模拟排序也能A
注意:其中Q.end() 返回的是最后一个值的下一个位置, 所以需要减一。
#include <bits/stdc++.h>
using namespace std;
const int maxn = 500005;
int n,k;
multiset <int> Q;
int main()
{
int i,j,x,t;
while ( cin>>n>>k ) {
Q.clear();
multiset <int>::iterator it1,it2;
for ( i=0; i<n; i++ ) {
scanf("%d",&x);
Q.insert(x);
}
while ( k-- ) {
it2 = --Q.end(); // 注意这里的--
t = *it2;
Q.erase(it2);
Q.insert(t-1);
it1 = Q.begin();
t = *it1;
Q.erase(it1);
Q.insert(t+1);
}
cout << *(--Q.end()) - *Q.begin() << endl;
}
return 0;
}