两个堆栈实现队列

2019独角兽企业重金招聘Python工程师标准>>>

算法很简单
如果你不明白，找一堆台球和两个杯子，想想怎么把球按照放的顺序取出来就行了。
整存整取思想

package queue

import java.util.*

fun main(args: Array) {
    var twoStackQueue: TwoStackQueue = TwoStackQueue()
    for (i in 0..9) {
        twoStackQueue.push(i)
    }
    for (i in 0..5) {
        println("print ${twoStackQueue.pop()}")
    }
    for (i in 10..19) {
        twoStackQueue.push(i)
    }
    while (!twoStackQueue.isEmpty())
        println("print ${twoStackQueue.pop()}")
}

/**
 * stack1是数据接收器, stack2是中转站
 * 每次put数据都放到stack1中,
 * 每次pop从stack2中读取, stack2如果没了,他会从stack1把数据取过来
 */
class TwoStackQueue {
    val stack1 = Stack()
    val stack2 = Stack()

    fun isEmpty(): Boolean {
        return stack1.isEmpty() && stack2.isEmpty()
    }

    fun push(t: T) {
        stack1.push(t)
    }

    fun pop(): T {
        if (stack2.isEmpty()) {
            while (!stack1.isEmpty()) {
                stack2.push(stack1.pop())
            }
        }
        return stack2.pop()
    }
}

转载于:https://my.oschina.net/sfshine/blog/2908535