"Coding Interview Guide" -- 设计一个有getMin功能的栈

 1 import java.util.Stack;
 2 
 3 class MyStack
 4 {
 5     private Stack<Integer> dataStack;
 6     private Stack<Integer> minStack;
 7 
 8     public MyStack()
 9     {
10     　　dataStack = new Stack<>();
11     　　minStack = new Stack<>();
12     }
13 
14     public void push(int data)
15     {
16     　　dataStack.push(data);
17     　　if(minStack.empty())    // 辅助栈为空或者栈顶元素小于或等于data时将data入栈，说明将data压入dataStack中后，当前dataStack的最小元素就是data，所以要更新minStack
18     　　{
19     　　　　minStack.push(data);
20     　　}
21     　　else if(data <= minStack.peek())
22　　　　　{
23             minStack.push(data);
24     　　}
25     }
26 
27     public int pop()
28     {
29     　　if(dataStack.empty())
30     　　{
31             throw new RuntimeException("fail to pop, stack is empty.");
32     　　}
33     　　else
34     　　{
35             int data = dataStack.pop();
36             if(data == getMin())          // 如果dataStack栈弹出的元素是当前dataStack中的最小元素，则弹出操作之后，需要对minStack栈进行更新
37             {
38                 minStack.pop();
39             }
40         　　return data;
41     　　}
42     }
43     
44     public int getMin()
45     {
46         if(minStack.empty())
47         {
48             throw new RuntimeException("fail to getMin, stack is empty.");
49     　　}
50     　　return minStack.peek();
51     }
52 
53     public int size()
54     {
55     　　return dataStack.size();
56     }
57 
58     public boolean isEmpty()
59     {
60         return dataStack.empty();
61     }
62 }