剑指 Offer 09. 用两个栈实现队列
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
解题思路:定义两个栈inStack和outStack。入队操作比较简单:队列中元素入队时就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
}
}