原创

剑指Offer面试题:19.包含Min函数的栈

一、题目:包含Min函数的栈

题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min、push及pop的时间复杂度都是O(1)。

  这里我们要实现的就是min、push以及pop三个方法:

    public class MinInStack<T> where T : struct
    {
        private Stack<T> dataStack;
        private Stack<T> minStack;

        public MinInStack()
        {
            this.dataStack = new Stack<T>();
            this.minStack = new Stack<T>();
        }

        public bool IsEmpty()
        {
            return this.dataStack.Count == 0;
        }

        public T Top()
        {
            return this.dataStack.Peek();
        }

        public void Push(T item)
        {
        }

        public T Pop()
        {
        }

        public T Min()
        {
        }
    }

二、解题思路

2.1 核心步骤

  把每次的最小元素(之前的最小元素和新压入栈的元素两者的较小值)都保存起来放到另外一个辅助栈里。下图展示了栈内压入3、4、2、1之后接连两次弹出栈顶数字再压入0时,数据栈、辅助栈和最小值的状态。

  从表中我们可以看出,如果每次都把最小元素压入辅助栈,那么就能保证辅助栈的栈顶一直都是最小元素

2.2 代码实现

  (1)Push方法

    public void Push(T item)
    {
        // 把新元素添加到数据栈
        dataStack.Push(item);
        // 当新元素比之前的最小元素小时,把新元素插入辅助栈里;
        // 否则把之前的最小元素重复插入辅助栈里
        if (minStack.Count == 0 || item.CompareTo(minStack.Peek()) < 0)
        {
            minStack.Push(item);
        }
        else
        {
            minStack.Push(minStack.Peek());
        }
    }

  (2)Pop方法

    public T Pop()
    {
        T item = dataStack.Pop();
        if(minStack.Count > 0)
        {
            minStack.Pop();
        }

        return item;
    }

  (3)Min方法

    public T Min()
    {
        return minStack.Peek();
    }

三、单元测试

3.1 测试用例

    [TestMethod]
    public void MinTest1()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        Assert.AreEqual(stack.Min(),3);
    }

    [TestMethod]
    public void MinTest2()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        stack.Push(4);
        Assert.AreEqual(stack.Min(), 3);
    }

    [TestMethod]
    public void MinTest3()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        stack.Push(4);
        stack.Push(2);
        Assert.AreEqual(stack.Min(), 2);
    }

    [TestMethod]
    public void MinTest4()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        stack.Push(4);
        stack.Push(2);
        stack.Push(3);
        Assert.AreEqual(stack.Min(), 2);
    }

    [TestMethod]
    public void MinTest5()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        stack.Push(4);
        stack.Push(2);
        stack.Push(3);
        stack.Pop();
        Assert.AreEqual(stack.Min(), 2);
    }

    [TestMethod]
    public void MinTest6()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        stack.Push(4);
        stack.Push(2);
        stack.Push(3);
        stack.Pop();
        stack.Pop();
        Assert.AreEqual(stack.Min(), 3);
    }

    [TestMethod]
    public void MinTest7()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        stack.Push(4);
        stack.Push(2);
        stack.Push(3);
        stack.Pop();
        stack.Pop();
        stack.Pop();
        Assert.AreEqual(stack.Min(), 3);
    }

    [TestMethod]
    public void MinTest8()
    {
        MinInStack<int> stack = new MinInStack<int>();
        stack.Push(3);
        stack.Push(4);
        stack.Push(2);
        stack.Push(3);
        stack.Pop();
        stack.Pop();
        stack.Pop();
        stack.Push(0);
        Assert.AreEqual(stack.Min(), 0);
    }

3.2 测试结果

  (1)测试通过情况

  (2)代码覆盖率

 

正文到此结束
本文目录