[剑指Offer] Day 01

剑指 Offer 09. 用两个栈实现队列

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )

解题思路:定义两个栈inStackoutStack。入队操作比较简单:队列中元素入队时就push到inStack中。出队列时统一"倒"到outStack中,顺序上做了一次调换要考虑以下几种情况:1.如果outStack中还有元素,直接pop即可;2.如果outStack没有元素,inStack也没有元素了,表明队列已经为空返回-1;3.如果outStack没有元素,inStack仍然有元素。先将所有元素"倒"到outStack中,然后弹出顶部元素即可。

class CQueue() {
    val inStack = Stack<Int>()
    val outStack = Stack<Int>()

    fun appendTail(value: Int) {
        inStack.push(value)
    }

    fun deleteHead(): Int {
        if (outStack.isEmpty().not()) {
            return outStack.pop()
        }

        if (inStack.isEmpty()) return -1

        while (inStack.isEmpty().not()) {
            outStack.push(inStack.pop())
        }

        return outStack.pop()

    }

}

剑指 Offer 30. 包含min函数的栈

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

解题思路:定义两个栈(一个栈存放压入的元素,一个栈存放每次更新后的最小值)和一个变量记录当前最小值。每次压入一个元素,都判断是否小于等于当前最小值,如果满足就将其存到最小数栈。要注意的是「等于」的情况也需要压入。每次弹出一个元素,就和最小值栈的栈顶比较下是否相同,如果相同则弹出一个元素。此时需要更新当前最小值变量。

class MinStack() {

    /** initialize your data structure here. */
    val stack = Stack<Int>()
    val minStack = Stack<Int>()
    var curMin = Int.MAX_VALUE
    
    fun push(x: Int) {
        stack.push(x)
        if (x <= curMin) {
            curMin = x
            minStack.push(x)
        }
    }

    fun pop() {
        var popNum = stack.pop()
        if (curMin == popNum) {
            minStack.pop()
            if (minStack.isEmpty().not()) {
                curMin = minStack.peek()
            } else {
                curMin = Int.MAX_VALUE
            }
        }

    }

    fun top(): Int {
        return stack.peek()
    }

    fun min(): Int {
        return curMin
    }

}
展示评论