设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) – 将元素 x 推入栈中。
pop() – 删除栈顶的元素。
top() – 获取栈顶元素。
getMin() – 检索栈中的最小元素。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.
思路:因为要在常数时间内得到最小值,所以无法用查找的方法来解决,需要用另外的一个栈来存储最小元素,当最小的元素入栈时,最小值栈也入栈,最小元素出栈时,最小值栈也出栈,这样最小值栈的栈顶便是最小元素。
首先是用数组存储,缺点是长度不可变,只能在构造时开辟足够的空间
public class MinStack {
int i=0;
public int size=0;
public int[] stack;
public int[] _minstack;
/** initialize your data structure here. */
public MinStack() {
stack=new int[10000];
_minstack=new int[10000];
}
public void Push(int x)
{
if(size==0)
stack[0]=x;
for(i=size;i>=0;i--)
{
stack[i+1]=stack[i];
}
stack[0]=x;
if(size==0)
_minstack[0]=x;
else if(x<=(int)_minstack[0]){
for(int i=size;i>=0;i--)
{
_minstack[i+1]=_minstack[i];
}
_minstack[0]=x;
}
size++;
}
public void Pop()
{
if(stack[0]==_minstack[0])
{
for( i=0;i<size;i++)
{
_minstack[i]=_minstack[i+1];
}
//stack[i+1]=null;
}
for( i=0;i<size;i++)
{
stack[i]=stack[i+1];
}
//stack[i+1]=null;
size--;
}
public int Top()
{
return (int)stack[0];
}
public int GetMin()
{
return (int)_minstack[0];
}
}
之后改用动态数组ArrayList存储,就不用担心数组长度不够的问题,入栈出栈也可以用动态数组的方法来写,要注意访问元素的时候需要强制转换成int类型,否则会出错。
public class MinStack {
public int size = 0;
public ArrayList stack;
public ArrayList _minstack;
/** initialize your data structure here. */
public MinStack()
{
stack = new ArrayList();
_minstack = new ArrayList();
}
public void Push(int x)
{
if (size == 0)
stack.Add(x);
else
{
stack.Insert(0, x);
}
if (size == 0)
_minstack.Add(x);
if (x <=(int) _minstack[0])
{
_minstack.Insert(0, x);
}
size++;
}
public void Pop()
{
if (size == 1)
{
stack.Clear();
_minstack.Clear();
size--;
}
else
{
if ((int)stack[0] == (int)_minstack[0])
{
_minstack.RemoveAt(0);
}
stack.RemoveAt(0);
size--;
}
}
public int Top()
{
return (int)stack[0];
}
public int GetMin()
{
return (int)_minstack[0];
}
}
时间复杂度:O(1)空间复杂度:O(n)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/min-stack
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。