131 lines
3.5 KiB
Java
131 lines
3.5 KiB
Java
|
//设计一个最大栈数据结构，既支持栈操作，又支持查找栈中最大元素。
|
|||
|
|
//
|
|||
|
|
// 实现 MaxStack 类：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// MaxStack() 初始化栈对象
|
|||
|
|
// void push(int x) 将元素 x 压入栈中。
|
|||
|
|
// int pop() 移除栈顶元素并返回这个元素。
|
|||
|
|
// int top() 返回栈顶元素，无需移除。
|
|||
|
|
// int peekMax() 检索并返回栈中最大元素，无需移除。
|
|||
|
|
// int popMax() 检索并返回栈中最大元素，并将其移除。如果有多个最大元素，只要移除 最靠近栈顶 的那个。
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 示例：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//输入
|
|||
|
|
//["MaxStack", "push", "push", "push", "top", "popMax", "top", "peekMax", "pop",
|
|||
|
|
// "top"]
|
|||
|
|
//[[], [5], [1], [5], [], [], [], [], [], []]
|
|||
|
|
//输出
|
|||
|
|
//[null, null, null, null, 5, 5, 1, 5, 1, 5]
|
|||
|
|
//
|
|||
|
|
//解释
|
|||
|
|
//MaxStack stk = new MaxStack();
|
|||
|
|
//stk.push(5); // [5] - 5 既是栈顶元素，也是最大元素
|
|||
|
|
//stk.push(1); // [5, 1] - 栈顶元素是 1，最大元素是 5
|
|||
|
|
//stk.push(5); // [5, 1, 5] - 5 既是栈顶元素，也是最大元素
|
|||
|
|
//stk.top(); // 返回 5，[5, 1, 5] - 栈没有改变
|
|||
|
|
//stk.popMax(); // 返回 5，[5, 1] - 栈发生改变，栈顶元素不再是最大元素
|
|||
|
|
//stk.top(); // 返回 1，[5, 1] - 栈没有改变
|
|||
|
|
//stk.peekMax(); // 返回 5，[5, 1] - 栈没有改变
|
|||
|
|
//stk.pop(); // 返回 1，[5] - 此操作后，5 既是栈顶元素，也是最大元素
|
|||
|
|
//stk.top(); // 返回 5，[5] - 栈没有改变
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 提示：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// -107 <= x <= 107
|
|||
|
|
// 最多调用 104 次 push、pop、top、peekMax 和 popMax
|
|||
|
|
// 调用 pop、top、peekMax 或 popMax 时，栈中 至少存在一个元素
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 进阶：
|
|||
|
|
//
|
|||
|
|
//
|
|||
|
|
// 试着设计解决方案：调用 top 方法的时间复杂度为 O(1) ，调用其他方法的时间复杂度为 O(logn) 。
|
|||
|
|
//
|
|||
|
|
// Related Topics 栈 设计 链表 双向链表 有序集合
|
|||
|
|
// 👍 70 👎 0
|
|||
|
|
|
|||
|
|
package leetcode.editor.cn;
|
|||
|
|
|
|||
|
|
import java.util.Stack;
|
|||
|
|
|
|||
|
|
//716:最大栈
|
|||
|
|
public class MaxStack {
|
|||
|
|
public static void main(String[] args) {
|
|||
|
|
//测试代码
|
|||
|
|
// Solution solution = new MaxStack().new Solution();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//力扣代码
|
|||
|
|
//leetcode submit region begin(Prohibit modification and deletion)
|
|||
|
|
class MaxStack {
|
|||
|
|
Stack<Integer> stack;
|
|||
|
|
Stack<Integer> max;
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* initialize your data structure here.
|
|||
|
|
*/
|
|||
|
|
public MaxStack() {
|
|||
|
|
stack = new Stack<>();
|
|||
|
|
max = new Stack<>();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void push(int x) {
|
|||
|
|
stack.push(x);
|
|||
|
|
if(max.isEmpty()||x>max.peek()){
|
|||
|
|
max.push(x);
|
|||
|
|
}else{
|
|||
|
|
max.push(max.peek());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public int pop() {
|
|||
|
|
max.pop();
|
|||
|
|
return stack.pop();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public int top() {
|
|||
|
|
return stack.peek();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public int peekMax() {
|
|||
|
|
return max.peek();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public int popMax() {
|
|||
|
|
int num = max.peek();
|
|||
|
|
Stack<Integer> temp = new Stack<>();
|
|||
|
|
while (stack.peek()!=num){
|
|||
|
|
temp.push(pop());
|
|||
|
|
}
|
|||
|
|
pop();
|
|||
|
|
while (!temp.isEmpty()){
|
|||
|
|
push(temp.pop());
|
|||
|
|
}
|
|||
|
|
return num;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* Your MaxStack object will be instantiated and called as such:
|
|||
|
|
* MaxStack obj = new MaxStack();
|
|||
|
|
* obj.push(x);
|
|||
|
|
* int param_2 = obj.pop();
|
|||
|
|
* int param_3 = obj.top();
|
|||
|
|
* int param_4 = obj.peekMax();
|
|||
|
|
* int param_5 = obj.popMax();
|
|||
|
|
*/
|
|||
|
|
//leetcode submit region end(Prohibit modification and deletion)
|
|||
|
|
|
|||
|
|
}
|