go的container/list 包里面带的list,有在链表头部和尾巴插入元素,和获取头部和尾部元素的操作,因此我们可以用list来实现栈和队列的功能。
import (
"container/list"
"fmt"
)
func StackTest() {
stack := list.New()
for i := 0; i < 5; i++ {
stack.PushBack(i) // 入栈
}
for {
if stack.Len() == 0 {
break
}
ele := stack.Back() //取出链表最后一个元素
val := ele.Value.(int)
fmt.Printf("val: %d", val)
stack.Remove(ele)//将最后一个元素删除,出栈
}
}
func QueueTest() {
queue := list.New()
for i := 0; i < 5; i++ {
queue.PushBack(i) // 入队列
}
for {
if queue.Len() == 0 {
break
}
ele := queue.Front() // 取出链表第一个元素
val := ele.Value.(int)
fmt.Printf("val: %d", val)
queue.Remove(ele) // 将第一个元素删除,出队列
}
}
说明一个,list链表元素element存放的val是interface类型的,我们将其取出来时,需要做一个类型转换