剑指offer面试题30:包含min函数的栈

题目:设计包含min 函数的栈。

  • 定义栈的数据结构,要求添加一个min 函数,能够得到栈的最小元素。
  • 要求函数min、push 以及pop 的时间复杂度都是O(1)。

思路:定义两个栈,一个用来储存数据,一个用来储存最小值。

测试用例:

  • 新压入栈中的数字比之前的最小值大
  • 新压入栈中的数字比之前的最小值小
  • 弹出的数字不是最小元素
  • 弹出的数字是最小元素
public class test_thirty {
    Stack<Integer> stack_data = new Stack<>();
    Stack<Integer> stack_min = new Stack<>();

    public void push(int node) {
        stack_data.push(node);
        if (stack_min.isEmpty()){
            stack_min.push(node);
        }
        else if (node < min()){
            stack_min.push(node);
        }
        else stack_min.push(min());
    }

    public void pop() {
        stack_data.pop();
        stack_min.pop();
    }

    public int top() {
        return stack_data.peek();
    }

    public int min() {
        return stack_min.peek();
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_41163113/article/details/86537258
今日推荐