写在前面:有时看一看数据结构与算法题,加深对于数据结构的理解,同时也是开阔自己的思路
题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)
题目分析
这个题目中关键点在于实现 min函数
的时间复杂度是O(1),一般得到栈中的最小值是要遍历整个栈的,现在要求达到O(1)的时间复杂度,采取空间换时间的做法
- 定义节点,增加一个最小值属性(这个的意思是以当前节点为栈顶,栈中元素的最小值)
class Node{
int val;
int min;
Node next;
}
- 用一个辅助的栈来保存当前栈中最小元素的变化
以上两种思路都是用空间来换时间,主要是对第一种进行实现,其中有一个细节需要细细思索
元素出栈会影响到栈中的最小元素,这个怎么考虑?
ps:这个可以看节点中最小值的定义,有点类似动态规划,当前节点始终保存的是整个栈中的最小值
解题代码
lass MinStack {
/** initialize your data structure here. */
public MinStack() {
}
private Node head;
public void push(int x) {
if (head == null){
head = new Node(x, x, null);
}
else{
head = new Node(x, Math.min(head.min, x), head);
}
}
public void pop() {
head = head.next;
}
public int top() {
return head.val;
}
public int min() {
return head.min;
}
private class Node{
int val;
int min;
Node next;
public Node(int val,int min,Node next){
this.val = val;
this.min = min;
this.next = next; //这个地方是链表中的头插法,生成的可以看成一个栈
}
}
}
/**
* Your MinStack object will be instantiated and called as such:
* MinStack obj = new MinStack();
* obj.push(x);
* obj.pop();
* int param_3 = obj.top();
* int param_4 = obj.min();
*/