Go 用list实现stack和queue

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类型的,我们将其取出来时,需要做一个类型转换

你可能感兴趣的:(go)