用栈来实现队列的golang实现

使用栈实现队列的下列操作：

• push(x) -- 将一个元素放入队列的尾部。
• pop() -- 从队列首部移除元素。
• peek() -- 返回队列首部的元素。
• empty() -- 返回队列是否为空。

MyQueue queue = new MyQueue();

queue.push(1);
queue.push(2);  
queue.peek();  // 返回 1
queue.pop();   // 返回 1
queue.empty(); // 返回 false

说明：

• 你只能使用标准的栈操作 -- 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
• 你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。
• 假设所有操作都是有效的 （例如，一个空的队列不会调用 pop 或者 peek 操作）。

栈的思想是先进后出，那我们怎么实现利用栈来实现先进先出的队列呢？

我们先来看张图

进入队列的顺序为1》2》3，那我们就把这三个元素放进input栈中，再把input栈中的元素放进output栈里，那么我们可以看见，我们从output读取的数据就是1》2》3了

那么我们一个一个方法来看：

push：对于push方法，我们只需要把元素放进input栈中即可

pop：对于pop方法，我们要从output栈中获取栈顶元素，如果output栈中为空，那么我们就要从input栈中把元素都放进output栈中，再从output栈中取出栈顶元素即可，如果两个栈都是空的，那就啥事都不干

peek：对于peek方法，与pop方法基本是一样，只是读取而已，不用去除元素

empty：只需要判断input栈和output栈的容量合起来是不是不为空即可

核心代码：

package main

type MyQueue struct {
    InputStack  []int //输入栈
    OutputStack []int //输出栈
}

/** Initialize your data structure here. */
func Constructor() MyQueue {
    queue := MyQueue{[]int{}, []int{}}
    return queue
}

/** Push element x to the back of queue. */
func (this *MyQueue) Push(x int) {
    this.InputStack = append(this.InputStack, x)
}

/** Removes the element from in front of queue and returns that element. */
func (this *MyQueue) Pop() int {
    //当输出栈不为空的，那就直接取出输出栈顶元素
    if len(this.OutputStack) != 0 {
        x := this.OutputStack[len(this.OutputStack)-1]
        this.OutputStack = this.OutputStack[:len(this.OutputStack)-1]
        return x
    }
    //如果输出栈为空，但是输入栈不为空，那就把输入栈的元素放进输出栈，并去除栈顶元素
    if len(this.InputStack) != 0 {
        for i := len(this.InputStack); i > 0; i-- {
            this.OutputStack = append(this.OutputStack, this.InputStack[i-1])
            this.InputStack = this.InputStack[:len(this.InputStack)-1]
        }
        x := this.OutputStack[len(this.OutputStack)-1]
        this.OutputStack = this.OutputStack[:len(this.OutputStack)-1]
        return x
    }
    return 0 //两个栈都是空的
}

/** Get the front element. */
func (this *MyQueue) Peek() int {
    if len(this.OutputStack) != 0 {
        return this.OutputStack[len(this.OutputStack)-1]
    }
    if len(this.InputStack) != 0 {
        for i := len(this.InputStack); i > 0; i-- {
            this.OutputStack = append(this.OutputStack, this.InputStack[i-1])
            this.InputStack = this.InputStack[:len(this.InputStack)-1]
        }
        return this.OutputStack[len(this.OutputStack)-1]
    }
    return 0
}

/** Returns whether the queue is empty. */
func (this *MyQueue) Empty() bool {
    if len(this.InputStack)+len(this.OutputStack) == 0 {
        return true
    }
    return false
}

func main() {
    queue := Constructor()
    queue.Push(1)
    queue.Push(2)
    _ = queue.Peek()
    _ = queue.Pop()
    _ = queue.Empty()
}

 

转载于:https://www.cnblogs.com/TimLiuDream/p/10087768.html