描述
实现一个特殊功能的栈,在实现栈的基本功能的基础上,再实现返回栈中最小元素的操作。
输入描述:
第一行输入一个整数N,表示对栈进行的操作总数。
下面N行每行输入一个字符串S,表示操作的种类。
如果S为"push",则后面还有一个整数X表示向栈里压入整数X。
如果S为"pop",则表示弹出栈顶操作。
如果S为"getMin",则表示询问当前栈中的最小元素是多少。
输出描述:
对于每个getMin操作,输出一行表示当前栈中的最小元素是多少。
示例1
输入:
6
push 3
push 2
push 1
getMin
pop
getMin
输出:
1
2
解题思路:
设计两个栈,一个是普通栈,与正常栈没区别,记作stackData。另一个就是记录最小值的栈,记作stackMin。
import java.util.Scanner;
import java.util.Stack;
public class Main {
public static void main(String[] args) {
MyStack1 myStack=new MyStack1();
Scanner in = new Scanner(System.in);
int N=in.nextInt();//输入操作数N
for(int i=0;i<=N;i++){//进行N次输入
String S=in.nextLine();//进行N次输入,每次输入的S都进行下面的三个判断
if(S.contains("push")){
Integer newNum=new Integer(S.split(" ")[1]);//如果S为"push",则后面还有一个整数X表示向栈里压入整数X。
myStack.push(newNum);
}
if(S.equals("getMin")){//如果S为"getMin",则表示询问当前栈中的最小元素是多少。
int min = myStack.getMin();
System.out.println(min);
}
if(S.equals("pop")){//如果S为"pop",则表示弹出栈顶操作。
int min = myStack.pop();
}
}
}
}
class MyStack1{
private Stack<Integer> stackData;//用来正常存储的栈 普通栈
private Stack<Integer> stackMin;//用来记录每一步的最小值
public MyStack1(){
this.stackData=new Stack<Integer>();
this.stackMin=new Stack<Integer>();
}
public void push(int newNum){
if(this.stackMin.isEmpty()){ //如果stackMin栈是空,直接压入
this.stackMin.push(newNum);
}else if(newNum<=this.getMin()){ //如果插入的值比最小值小,压入stackMin栈
this.stackMin.push(newNum);
}
this.stackData.push(newNum);//正常压入 普通栈
}
public int pop(){
if(this.stackData.isEmpty()){
throw new RuntimeException("你的栈啥也没有!");
}
int value=this.stackData.pop(); //从普通栈取出栈顶元素记作value
if(value==this.getMin()){ //如果value等于最小值,就从stackMin栈取出来
this.stackMin.pop();
}
return value; //返回取出来的value
}
public int getMin(){
if(this.stackMin.isEmpty()){
throw new RuntimeException("你的栈啥也没有!");
}
return this.stackMin.peek();//stackMin栈的栈顶元素就是最小值
}
}