题解 | #设计getMin功能的栈#
设计getMin功能的栈
http://www.nowcoder.com/practice/05e57ce2cd8e4a1eae8c3b0a7e9886be
设计getMin功能的栈
- 创建两个栈 一个栈stack_data存所有数据,一个栈stack_min【栈顶最小】存实时最小值
- 进栈push: stack_min将每个数据与栈顶数据比较(空栈直接进来) 小于栈顶则进来
- 出栈pop: stack_data出栈数据和 stack_min栈顶比较(只有可能>=) =直接 stack_min直接出栈
- getMin 直接出实时最小 stack_min 栈顶元素
import java.util.*;
import java.io.*;
public class Main{
static Stack<Integer> stack_data = new Stack<>();
static Stack<Integer> stack_min = new Stack<>();
//栈空 或者 stack_min栈顶比较进栈数据小者存进来
public static void _push(int num){
if(stack_min.isEmpty()){
stack_min.push(num);
}else if(num <= stack_min.peek()){
stack_min.push(num);
}
stack_data.push(num);
}
//出栈 如果出栈 stack_data 的数据和 stack_min 相同 则stack_min出栈
public static void _pop(){
if(stack_data.isEmpty()){
System.out.println("栈空");
return;
}
int current = stack_data.pop();
if(current == stack_min.peek()){
stack_min.pop();
}
}
public static void main(String args[]) throws IOException{
BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
int n = Integer.valueOf(in.readLine());
for(int i = 0; i < n; i++){
String ins[] = in.readLine().split(" ");
String str = ins[0];
if("push".equals(str)){
int num = Integer.parseInt(ins[1]);
_push(num);
}else if("pop".equals(str)){
_pop();
}else{
System.out.println(stack_min.peek());
}
}
in.close();
}
}