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;
}