题目概述
- 算法说明
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。 - 测试用例
队列先进先出,输入: 1,2
输出: 1,2
解析&参考答案
- 解析
golang 中默认没有栈结构,因此此处直接使用两个list来模拟一个队列功能。 - 参考答案
vim jz25.go
package main
import "fmt"
var stack1 []int
var stack2 []int
func Push(node int) {
stack1 = append(stack1, node)
}
func Pop() int {
ret := 0
if len(stack2) == 0 {
for i := len(stack1) - 1; i >= 0; i-- {
stack2 = append(stack2, stack1[i])
}
stack1 = []int{}
ret = stack2[len(stack2)-1]
stack2 = stack2[:len(stack2)-1]
} else {
ret = stack2[len(stack2)-1]
stack2 = stack2[:len(stack2)-1]
}
return ret
}
func main() {
Push(1)
Push(2)
fmt.Println(Pop())
fmt.Println(Pop())
}
注意事项
- to add
说明
- 当前使用 go1.15.8
- 参考 牛客网--剑指offer
标题中jzn(n为具体数字)代表牛客网剑指offer系列第n号题目,例如 jz01 代表牛客网剑指offer中01号题目。
注意!!!
- 笔者最近在学习 golang,因此趁机通过数据结构和算法来进一步熟悉下go语言
- 当前算法主要来源于剑指 offer,后续会进一步补充 LeetCode 上重要算法,以及一些经典算法
- 此处答案仅为参考,不一定是最优解,欢迎感兴趣的读者在评论区提供更优解