原题地址 :hihoCoder
http://hihocoder.com/contest/hiho28/problem/1
输入
每个测试点(输入文件)有且仅有一组测试数据。
在一组测试数据中:
第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,表示在这一时刻,盒子中最重的糖果的重量。
我的c++代码,但是给了我WA,真是很莫名其妙
#include <vector>
#include <string>
#include <iostream>
using namespace std;
vector<long> lvec;
long sizeOfVec;
void add1(long index)
{
long temp = lvec[index];
while (index > 1 && temp > lvec[index / 2])
{
lvec[index] = lvec[index / 2];
index /= 2;
}
lvec[index] = temp;
}
void delete1(long index)
{
long temp = lvec[index];
while (index * 2 <= sizeOfVec)
{
long child = index * 2;
if (child < sizeOfVec && lvec[child] < lvec[child + 1]) child++;
if (temp < lvec[child])
{
lvec[index] = lvec[child];
index = child;
}
else{
break;
}
}
lvec[index] = temp;
}
int main()
{
long n = 0;
cin >> n;
sizeOfVec = 0;
lvec.push_back(-1);//堆从1开始,0位置放弃使用
string s;
for (long i = 0; i < n; i++)
{
cin >> s;
if (s == "A")
{
long temp = 0;
cin >> temp;
sizeOfVec++;
lvec.push_back(temp);
add1(sizeOfVec);
}
else{
cout << lvec[1] << endl;
lvec[1] = lvec[sizeOfVec];
sizeOfVec--;lvec.pop_back();
if(sizeOfVec!=0) delete(1);
}
}
return 0;
}