hihocoder 1105 题外话·堆 堆的应用

描述

小Ho有一个糖果盒子，每过一段时间小Ho都会将新买来的糖果放进去，同时他也会不断的从其中挑选出最大的糖果出来吃掉，但是寻找最大的糖果不是一件非常简单的事情，所以小Ho希望能够用计算机来他帮忙计算这个问题！

输入

每个测试点（输入文件）有且仅有一组测试数据。

在一组测试数据中：

第1行为1个整数N，表示需要处理的事件数目。

接下来的M行，每行描述一个事件，且事件类型由该行的第一个字符表示，如果为'A'，表示小Ho将一粒糖果放进了盒子，且接下来为一个整数W，表示这颗糖果的重量；如果为'T'，表示小Ho需要知道当前盒子中最重的糖果的重量是多少，在知道这个值之后，小Ho会将这颗糖果从盒子中取出并吃掉。

对于100%的数据，满足1<=N<=10^5, 1<=w<=10^5。<>

对于100%的数据，满足没有2颗糖果的重量是相同的，最开始的时候小Ho的糖果盒子是空的，且每次小Ho想要取出一颗糖果的时候盒子里一定至少有一颗糖果。

输出

在一组测试数据中：

对于每个类型为'T'的时间，输出1个整数W_MAX，表示在这一时刻，盒子中最重的糖果的重量。

样例输入

5
A 77751
A 1329
A 26239
A 80317
T

样例输出

80317

一共两种操作 放入和取出（MAX）的

最多有10W次操作 ，暴力肯定会超时。

我们可以将盒子理解为一个大顶堆，即父节点大于左右子节点。

1.每次放入糖果时往上维护堆

2.取出时模仿堆排序的算法 将根节点(max)输出并与最后一个节点交换  再维护堆

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int q[100005];
void HeapAdjust(int loc,int len)
{
    int pos=loc,leftchild=2*loc,rightchild=2*loc+1;
    if(loc<=len/2)
    {
        if(leftchild<=len&&q[leftchild]>q[pos])
            pos=leftchild;
        if(rightchild<=len&&q[rightchild]>q[pos])
            pos=rightchild;
        if(pos!=loc)
        {
            swap(q[pos],q[loc]);
            HeapAdjust(pos,len);
        }
    }
    return;
}
int main()
{
    int n,len=0,b;
    char a;
    scanf("%d",&n);
    getchar();
    for(int j=1;j<=n;j++)
    {
        cin>>a;
        if(a=='A')
        {
            scanf("%d",&q[++len]);
            int l=len/2;
            while(l!=0)                 //依次往上维护堆  
            {
                HeapAdjust(l,len);
                l=l/2;
            }
            getchar();
        }
        else if(a=='T')
        {
            printf("%d\n",q[1]);
            swap(q[1],q[len--]);
            HeapAdjust(1,len);          //维护
        }
    }
    return 0;
}