GYM 100694 E.SuperHyperMarket（set）

Description
超市有k家店，n个人依次来逛超市，第i个人有p[i]的消费水平，一个客人来到超市后会看看在每家店消费的最后两人的消费水平均值（如果只有一人那么均值就是这个人的消费水平，如果没有人就是0），这个人会选择均值最小的那家店进去消费，如果有几家店的均值都是最小的则选择编号最小的那家店进去，问这n个人选择消费的店铺
Input
第一行两个整数n和k表示人数和店数，之后n个整数p[i]表示第i个人的消费水平(1<=n,k<=1e5,1<=p[i]<=1e3)
Output
输出n个整数表示这n个人选择消费的店铺
Sample Input
5 2
1 3 1 2 3
Sample Output
1 2 1 1 2
Solution
开一个set记录二元组（i，j），i表示第j家店铺的消费均值，再开一个二元数组记录每家点最后两位顾客的消费水平，第k个人来之后把set里第一个元素（i，j）拿出来，因为这就是这个人的选择（i最小，i相同则j最小），然后把这个元素删除，更新第i家店最后两个人的消费水平得到新的消费均值jj，把原先最后一个人变成现在的倒数第二个人，第k个人变成最后一个，然后把新的（i，jj）插入到set里
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 111111
typedef pair<int,int>P;
set<P>s;
set<P>::iterator it;
P a[maxn];
int n,k,p[maxn],num[maxn];
int main()
{
    while(~scanf("%d%d",&n,&k))
    {
        s.clear();
        for(int i=1;i<=n;i++)scanf("%d",&p[i]);
        for(int i=1;i<=k;i++)s.insert(P(0,i)),a[i]=P(0,0),num[i]=0;
        for(int i=1;i<=n;i++)
        {
            it=s.begin();
            int pp=(*it).first,id=(*it).second;
            printf("%d%c",id,i==n?'\n':' ');
            a[id].first=a[id].second,a[id].second=p[i],num[id]++;
            s.erase(it);
            int temp=num[id]*(a[id].first+a[id].second);
            if(a[id].first==0)temp*=2;
            s.insert(P(temp,id));
        }
    }
    return 0;
}