题目描述:
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
思路:
题目要求找到最小值得时间复杂度是O(1),说明不能遍历数据集合,需要立马找到,那就只能从数据的push和pop入手,下面是我的思考过程:
1.首先是考虑加一个变量记录当前的最小值,在push的时候记录放入的数据的最小值,但是在pop出最小值的时候,就不知道当前栈内的最小值是多少了
2.所以每次push数据的时候,可以考虑增设一个数据集合按小到大排序, 每push一次数据,就将新数据放到有序集合里面,但要使集合仍然有序。而且在pop的时候,要将集合中的对应的一个元素去掉。
3.这样每次取最小值的时候直接取数据集合的头一个元素即可。
import java.util.Iterator;
import java.util.LinkedList;
import java.util.Stack;
public class Solution {
Stack<Integer> a = new Stack<Integer>();
LinkedList<Integer> minList = new LinkedList<Integer>();
public void push(int node) {
if(minList.size() == 0){
minList.add(node);
}else{
Integer temp = null;
Iterator<Integer> iter;
for(iter = minList.iterator(); iter.hasNext();){
temp = (Integer) iter.next();
if(temp >= node){
break;
}
}
if(temp < node){
//链表中找不到比它小的数,当前加入的是最大的,应加到末尾
minList.addLast(node);
}else{
minList.add(minList.indexOf(temp),node);
}
}
a.push(node);
}
public void pop() {
Integer temp = a.pop();
minList.removeFirstOccurrence(temp);//仅删除一个
}
public int top() {
return a.peek();
}
public int min() {
return minList.getFirst();
}
}