Stack (30)
难度: ⭐⭐⭐⭐
题目连接
题目描述
Stack is one of the most fundamental data structures, which is based on the principle of Last In First Out (LIFO). The basic operations include Push (inserting an element onto the top position) and Pop (deleting the top element). Now you are supposed to implement a stack with an extra operation: PeekMedian – return the median value of all the elements in the stack. With N elements, the median value is defined to be the (N/2)-th smallest element if N is even, or ((N+1)/2)-th if N is odd.
输入描述:
Each input file contains one test case. For each case, the first line contains a positive integer N (<= 105). Then N lines follow, each contains a command in one of the following 3 formats:
Push key
Pop
PeekMedian
where key is a positive integer no more than 105.
输出描述:
For each Push command, insert key into the stack and output nothing. For each Pop or PeekMedian command, print in a line the corresponding returned value. If the command is invalid, print “Invalid” instead.
输入例子:
17
Pop
PeekMedian
Push 3
PeekMedian
Push 2
PeekMedian
Push 1
PeekMedian
Pop
Pop
Push 5
Push 4
PeekMedian
Pop
Pop
Pop
Pop
输出例子:
Invalid
Invalid
3
2
2
1
2
4
4
5
3
Invalid
大意
给出三种操作:在栈的基础功能上再增加一个要求,能在任何时候够高效地取出栈里面大小排在中间的那个数。
分析
主要的难点是做一个能够随时取出中位数的数据结构,可以维护两个堆,使得任何时候第一个堆的大小不比第二个多于一,那么第一个堆的最大元素就是所需要的中位数。
MYCODE
#include<iostream>
#include<string.h>
#include<stdio.h>
#include<string>
#include<set>
#include<stack>
using namespace std;
class Myset{
multiset<int> s1, s2;
public