剑指Offer--【包含min函数的栈】--java

题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

知识回顾:
java中的栈Stack的基本使用和应用

定义  栈是一种只能在一端进行插入或删除操作的线性表。(先进后出表)

java中的Stack继承Vector

实例化

Stack stack=new Stack();
基本使用

判断是否为空stack.empty()

取栈顶值(不出栈)stack.peek()

进栈stack.push(Object);

出栈stack.pop();

代码如下:
import java.util.Stack;

扫描二维码关注公众号,回复: 5292344 查看本文章

public class Solution {
// 数据栈
Stack < Integer> dataStack = new Stack< Integer>();
// 辅助栈
Stack < Integer> minStack = new Stack< Integer>();
Integer temp = null;

// 每次压入数据栈数据与最小数据进行比较,如果更小,存入数据栈同时,存入辅助栈,并且最小值设置为该数据
public void push(int node)
{
dataStack.push(node);
if (minStack.isEmpty())
{
minStack.push(node);
temp = node;
} else
{
temp = minStack.peek();
if (node <= temp)
{
minStack.push(node);
temp = node;
} else
// 依然是原来那个数更小,继续存入该最小值
minStack.push(temp);
}
}

// 数据栈出栈的同时,辅助栈也出栈,保证辅助栈的栈顶一直是最小元素
public void pop()
{
dataStack.pop();
minStack.pop();
}

public int top()
{

return dataStack.peek();
}

public int min()
{
return minStack.peek();
}
}

运行结果:
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/weixin_43689040/article/details/87885486