题解 | #用两个栈实现队列#

题目

用两个栈来实现一个队列，使用n个元素来完成 n 次在队列尾部插入整数(push)和n次在队列头部删除整数(pop)的功能。 队列中的元素为int类型。保证操作合法，即保证pop操作时队列内已有元素。

数据范围： n\le1000n≤1000

要求：存储n个元素的空间复杂度为 O(n)O(n) ，插入与删除的时间复杂度都是 O(1)O(1)

不正经解法

这道题本质就是数组模拟栈，再用两个栈模拟队列，那么何必那么麻烦，直接一个数组模拟队列就完了，狗头保命

package main

var stack1 [] int
var stack2 [] int

func Push(node int) {
    stack1=append(stack1,node)
}

func Pop() int{
    //牛客标准答案
//     if len(stack2)==0{
//         for i:=0;i<len(stack1);i++{
//             stack2 = append(stack2,stack1[len(stack1)-1-i])
//         }
//         stack1 = []int{}
//     }
//     res:=stack2[len(stack2)-1]
//     stack2 = stack2[:len(stack2)-1]
//     return res
    //---------------------------------------
    //下面是不正经解法
	var res int
	if len(stack1)!=0{
		res = stack1[0]
		stack1 = stack1[1:]
	}
	return res
}