設計有一個getMin功能的棧（ java )

設計有一個getMin功能的棧

說明
本文來自左程雲老師所著的《程序員面試代碼指南》第一章
題目
實現一個特殊的棧，在實現棧的基本功能的基礎上，再實現返回棧中最小元素的操作。
要求
1.pop、push、 getMin操作的時間複雜度都是0(1)。
2.設計的棧類型可以使用現成的棧結構。
方法
在設計上使用兩個棧，一個棧用來保存當前棧中的元素，其功能和一個正常的棧
沒有區別，這個棧記爲stackData; 另一個棧用於保存每一步的最小值，這個棧記爲stackMin。
舉例
stackData存放3 4 5 1 2 1
則stackMin存放3 1 1
代碼實現

import java.util.Scanner;
public class MyStack {
 
 private Stack<Integer> stackData;
 private Stack<Integer> stackMin;
 
 public static void main(String[] args) {
 Scanner cs = new Scanner(System.in);
 MyStack stack = new MyStack();
 System.out.println("輸入數據個數：");
 int n = cs.nextInt();
 for(int i=0;i<n;i++) {
  int temp = cs.nextInt();
  stack.push(temp);
 }
 stack.pop();  //只爲測試pop函數
 System.out.println(stack.getMin());
}
 
 public MyStack()
 {
  stackData = new Stack<Integer>();
  stackMin = new Stack<Integer>();
 }
 
 void push(int m) {
  stackData.push(m);
  if(stackMin.empty()) stackMin.push(m);
  else {
   if(stackMin.peek()>=m) 
    stackMin.push(m); 
    //stackMin棧頂元素大於等於m時加入
    }
 }
   //此後當有元素出棧時，stackData正常出棧，如若出棧元素等於stackMin的棧頂，
    //則stackMin棧頂一同出棧，保證實時更新stackMin

 void pop() {
  if(stackData.empty()) 
  throw new RuntimeException("Your empty is empty!");
  int a = stackData.pop();
  if(a == stackMin.peek()) stackMin.pop();
 }
 
 int getMin() {
  if(stackData.empty()) 
  throw new RuntimeException("Your empty is empty!");
  int a = stackMin.peek();
  return a;
 }
}

說明：只爲記錄自己每日學習，加油 ~