codevs 2830 蓬莱山辉夜

#include<iostream>
#include<cstdio>
#include<algorithm>
struct node
{
    int a,b,c;//a编号，b已花费时间，c等待时间 
}e[10001];
using namespace std;
int n,s,i,l[10010];//s堆heap中元素的个数，l用于存放时间相同的少女的编号 
char c[1001];
node heap[10010];//堆 
bool cmp(node k,node p)
{
    return k.b<p.b;
}
void insert(node k)//向堆中加入元素 
{
    s++;
    heap[s]=k;
    int p=s;
    while(p>1&&k.b<heap[p/2].b)
    {
        heap[p]=heap[p/2];
        p/=2;
    }
    heap[p]=k;
}
void heapify(int k)//维护小根堆 
{
    int left=k*2,right=k*2+1;
    int minn=k;
    if(left<=s) minn=heap[minn].b<heap[left].b ? minn:left;
    if(right<=s) minn=heap[minn].b<heap[right].b ? minn:right;
    if(minn!=k)
    {
        swap(heap[minn],heap[k]);
        heapify(minn);
    }
}
node get()//取出堆中第一个元素 
{
    node k=heap[1];
    heap[1]=heap[s];
    s--;
    heapify(1);
    return k;
}
int main()
{
    cin>>c;
    while(c[0]!=‘#‘)
    {
        i++;
        scanf("%d%d",&e[i].a,&e[i].b);
        e[i].c=e[i].b;
        insert(e[i]);
        cin>>c;
    }
    scanf("%d",&n);
    while(n)
    {
        bool ok=false; 
        node h=get();//取出堆中的第一个元素 
        int o=h.a;//这次要输出的元素的编号 
        int su=0;//相同时间的少女的个数 
        while(h.b==heap[1].b) //有少女在相同时间被访问到 
        {
            ok=true;
            l[++su]=h.a;//存进去 
            h.b+=h.c;//加上等待时间 
            insert(h);//再放回堆里 
            h=get();//再取出堆得第一个元素 
            o=min(o,h.a);
        }
        l[++su]=h.a;
        h.b+=h.c;
        insert(h);
//如过存在几个少女同一时间被访问到的情况，那根据上面的while，最后一个同一时间的少女没有取出来放入l数组，所以要加进去 
//如果不存在几个少女同一时间被访问的情况，语句l[++su]=h.a执行没有影响，实际有作用的是后两句        
        if(ok)
        {
            sort(l+1,l+su+1);//按字典序输出 
            for(int i=1;i<=su;i++)
            printf("%d\n",l[i]);
            n-=su;
            continue;
        }
        printf("%d\n",o);//没有少女在同一时间被访问，直接输出编号 
        n--;
    }
}