mirror of
https://github.com/youngyangyang04/leetcode-master.git
synced 2025-07-09 11:34:46 +08:00
Update0232.用栈实现队列 优化Go代码实现
使用切片模拟栈的实现,即 MyStack
This commit is contained in:
@ -235,52 +235,77 @@ class MyQueue:
|
|||||||
### Go:
|
### Go:
|
||||||
|
|
||||||
```Go
|
```Go
|
||||||
type MyQueue struct {
|
// 通过切片实现一个栈
|
||||||
stackIn []int //输入栈
|
// 由于只是辅助实现算法题目,因此不做异常情况处理
|
||||||
stackOut []int //输出栈
|
type MyStack []int
|
||||||
|
|
||||||
|
func (s *MyStack) Push(v int) {
|
||||||
|
*s = append(*s, v)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *MyStack) Pop() int {
|
||||||
|
val := (*s)[len(*s)-1]
|
||||||
|
*s = (*s)[:len(*s)-1]
|
||||||
|
return val
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *MyStack) Peek() int {
|
||||||
|
return (*s)[len(*s)-1]
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *MyStack) Size() int {
|
||||||
|
return len(*s)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *MyStack) Empty() bool {
|
||||||
|
return s.Size() == 0
|
||||||
|
}
|
||||||
|
|
||||||
|
// ---------- 分界线 ----------
|
||||||
|
|
||||||
|
type MyQueue struct {
|
||||||
|
stackIn *MyStack
|
||||||
|
stackOut *MyStack
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
func Constructor() MyQueue {
|
func Constructor() MyQueue {
|
||||||
return MyQueue{
|
return MyQueue {
|
||||||
stackIn: make([]int, 0),
|
stackIn: &MyStack{},
|
||||||
stackOut: make([]int, 0),
|
stackOut: &MyStack{},
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// 往输入栈做push
|
|
||||||
func (this *MyQueue) Push(x int) {
|
func (this *MyQueue) Push(x int) {
|
||||||
this.stackIn = append(this.stackIn, x)
|
this.stackIn.Push(x)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 在输出栈做pop,pop时如果输出栈数据为空,需要将输入栈全部数据导入,如果非空,则可直接使用
|
|
||||||
func (this *MyQueue) Pop() int {
|
func (this *MyQueue) Pop() int {
|
||||||
inLen, outLen := len(this.stackIn), len(this.stackOut)
|
this.fillStackOut()
|
||||||
if outLen == 0 {
|
return this.stackOut.Pop()
|
||||||
if inLen == 0 {
|
|
||||||
return -1
|
|
||||||
}
|
|
||||||
for i := inLen - 1; i >= 0; i-- {
|
|
||||||
this.stackOut = append(this.stackOut, this.stackIn[i])
|
|
||||||
}
|
|
||||||
this.stackIn = []int{} //导出后清空
|
|
||||||
outLen = len(this.stackOut) //更新长度值
|
|
||||||
}
|
|
||||||
val := this.stackOut[outLen-1]
|
|
||||||
this.stackOut = this.stackOut[:outLen-1]
|
|
||||||
return val
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func (this *MyQueue) Peek() int {
|
func (this *MyQueue) Peek() int {
|
||||||
val := this.Pop()
|
this.fillStackOut()
|
||||||
if val == -1 {
|
return this.stackOut.Peek()
|
||||||
return -1
|
|
||||||
}
|
|
||||||
this.stackOut = append(this.stackOut, val)
|
|
||||||
return val
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
func (this *MyQueue) Empty() bool {
|
func (this *MyQueue) Empty() bool {
|
||||||
return len(this.stackIn) == 0 && len(this.stackOut) == 0
|
return this.stackIn.Empty() && this.stackOut.Empty()
|
||||||
|
}
|
||||||
|
|
||||||
|
// fillStackOut 填充输出栈
|
||||||
|
func (this *MyQueue) fillStackOut() {
|
||||||
|
if this.stackOut.Empty() {
|
||||||
|
for !this.stackIn.Empty() {
|
||||||
|
val := this.stackIn.Pop()
|
||||||
|
this.stackOut.Push(val)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user