L3-002. 堆栈

时间限制

200 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

大家都知道“堆栈”是一种“先进后出”的线性结构，基本操作有“入栈”（将新元素插入栈顶）和“出栈”（将栈顶元素的值返回并从堆栈中将其删除）。现请你实现一种特殊的堆栈，它多了一种操作叫“查中值”，即返回堆栈中所有元素的中值。对于N个元素，若N是偶数，则中值定义为第N/2个最小元；若N是奇数，则中值定义为第(N+1)/2个最小元。

输入格式：

输入第一行给出正整数N（<= 10⁵）。随后N行，每行给出一个操作指令，为下列3种指令之一：

Push key
Pop
PeekMedian

其中Push表示入栈，key是不超过10⁵的正整数；Pop表示出栈；PeekMedian表示查中值。

输出格式：

对每个入栈指令，将key入栈，并不输出任何信息。对每个出栈或查中值的指令，在一行中打印相应的返回结果。若指令非法，就打印“Invalid”。

输入样例：

17PopPeekMedianPush 3PeekMedianPush 2PeekMedianPush 1PeekMedianPopPopPush 5Push 4PeekMedianPopPopPopPop

输出样例：

InvalidInvalid322124453Invalid

思路分析：

用一个栈模拟实现栈的输入输出即可，关键在于如何查询中值。可以利用分块思想处理，先确定待查询的数在哪一块儿，再在块儿内进行查询即可。

题解：

#include <cstdio>#include <stack>#include <cstring>using namespace std;const int MAX = 100001;const int MAXN = 317;int n, block[MAXN] = {0}, table[MAX] = {0};char order[11];stack<int> s;bool cmp(int a, int b){return a < b;}int main(){scanf("%d", &n);while(n--){scanf("%s", order);if(strcmp(order, "Pop") == 0){if(!s.empty()){int k = s.top();printf("%d\n", k);s.pop();block[k/(MAXN-1)]--;table[k]--;}else printf("Invalid\n");}else if(strcmp(order, "PeekMedian") == 0){if(s.empty()){printf("Invalid\n");continue;}int k, idx = 0, index = 0;if(s.size() % 2 == 0) k = s.size()/2;else k = (s.size()+1)/2;int sum = 0;while(sum + block[idx] < k){sum += block[idx++];}index = idx * (MAXN-1);while(sum + table[index] < k){sum += table[index++];}printf("%d\n", index);}else{int k;scanf("%d", &k); s.push(k);table[k]++;block[k/(MAXN-1)]++;}}return 0;}