Honk's pool[STL multiset]

目录

• 题目地址
• 题干
• 代码和解释

题目地址

Honk's pool（The Preliminary Contest for ICPC Asia Shenyang 2019 ）

题干

代码和解释

本题使用了STL multiset，它与 set 一样可以对元素自动排序，而与 set 不同的是能存多个值相同的元素。

c++代码如下：

#include<bits/stdc++.h>
using namespace std;
int main()
{
	int n,k,x;
	int i;
	int tmp;
	while(~scanf("%d%d",&n,&k)){
		multiset<int> SS;
		multiset<int>::iterator it1,it2;
		for(i=0;i<n;i++){
			scanf("%d",&x);
			SS.insert(x);
		}
		for(i=0;i<k;i++){
			it1=SS.begin();
			tmp=*it1;//不能直接改变元素值，否则会打乱顺序。应先删除旧元素，再插入新元素
			tmp++;
			SS.erase(it1);
			SS.insert(tmp); 

			it2=SS.end();
			it2--;
			tmp=*it2;
			tmp--;
			SS.erase(it2);
			SS.insert(tmp);

			//for(it1=SS.begin();it1!=SS.end();it1++)
			//	printf("%d ",*it1);
			//printf("\n");
		}
		it1=SS.begin();
		it2=SS.end();
		it2--;
		printf("%d\n",(*it2)-(*it1));
	}
	return 0;
}