设计getMin功能的栈

描述

实现一个特殊功能的栈，在实现栈的基本功能的基础上，再实现返回栈中最小元素的操作。

输入描述：

第一行输入一个整数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栈的栈顶元素就是最小值
    }
}