百炼 4078:实现堆结构（模拟堆）

描述

定义一个数组，初始化为空。在数组上执行两种操作：

1、增添1个元素，把1个新的元素放入数组。

2、输出并删除数组中最小的数。

使用堆结构实现上述功能的高效算法。

输入第一行输入一个整数n，代表操作的次数。
每次操作首先输入一个整数type。
当type=1，增添操作，接着输入一个整数u，代表要插入的元素。
当type=2，输出删除操作，输出并删除数组中最小的元素。
1<=n<=100000。输出每次删除操作输出被删除的数字。样例输入

4
1 5
1 1
1 7
2

样例输出

1

提示每组测试数据的复杂度为O(nlogn)的算法才能通过本次，否则会返回TLE(超时)

需要使用最小堆结构来实现本题的算法

 

没有注意到cnt=0的情况，参考了别人代码才AC

 

#include<cstring>
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;

int k;
int cnt=0;
int num[100005];

bool cmp(int a,int b)
{
    return a>b;
}

int main()
{

    memset(num,0,sizeof(num));
    cnt=0;
    scanf("%d",&k);
    for(int i=0; i<k; i++)
    {
        int m;
        scanf("%d",&m);
        if(m==1)
        {
            int temp;
            scanf("%d",&temp);

            num[++cnt]=temp;

            int i=cnt;

            while(i/2>0)
            {
                if(temp<num[i/2])
                {
                    num[i]=num[i/2];
                    i=i/2;
                }
                else
                {
                    break;
                }
            }
            num[i]=temp;
        }
        else
        {
            int minn=num[1];
            int temp=num[cnt];
            printf("%d\n",minn);
            cnt--;
            if(cnt==0)
                continue;
            int i=1;
            while(i*2<=cnt)
            {
                int a;
                if(i*2+1<=cnt&&num[i*2]>num[i*2+1]&&temp>num[i*2+1])
                {
                    a=num[i*2+1];
                    num[i]=a;
                    i=i*2+1;
                    continue;
                }

                if(temp>num[i*2])
                {
                    a=num[i*2];
                    num[i]=a;
                    i=i*2;
                    continue;
                }

                else
                {
                    break;
                }
            }
            num[i]=temp;


        }
    }

    return 0;
}