编程之美 set 10 队列中取最大值操作问题

题目

假设有这样一个拥有三个操作的队列

1. Enqueue(v)

2. Dequeue()

3. MaxEle()

请设计一种数据结构和算法, 让 MAXELE 操作的时间复杂度尽可能的低

 

思路

1. 刚看到题, 以为这就是 stack 求 min 的逆过程, 着实是理解有误

2. 比较优雅的解法是利用两个堆栈来模拟队列, 模拟的方法以前是做过习题的, 不必多说

3. 重点是 stack 的实现. stack 有 4 个私有变量

Type stackItem[MAXN]; // store real numbers
int stackTop;	// point to top of stack
int link2NextMaxItem[MAXN]; // another stack storing index of max number
int maxStackItemIndex; // point to max number in stack

　

假如队列的一次输入  1 3 2 4 2

那么对应堆栈的输入  1 3 2 4 2

　　　　　　　　　  0 1 -1 3 -1

当队列需要输出时, stackA 的数据转到 stackB 中, 数据变成 2 4 2 3 1 (0 1 -1 -1 -1)

注意, 最后输出的时候, 要比较 stackA.max() 和 stackB.max()