GO语言栈(链表实现)

132 阅读1分钟

栈,先进后出的规则

package main

import "fmt"

//Student is a test struct
type Student struct {
	base  *Student
	top   *Student
	idnum int
	Next  *Student
}

func push(head *Student, arr []int, top *Student) *Student { //向栈中添加新的元素
	for _, num := range arr {
		lstu := Student{idnum: num, Next: nil}
		top.Next = &lstu //将数组中的元素的地址放到上一个元素的next指针位置
		top = &lstu
	}
	return top //返回头节点,这里可以返回,可以不返回头节点,使用作为参数传进来的头节点也可以,在链表之后添加新的元素对头节点不影响,但是如果不返回头节点,需要在声明函数时,写无返回值
}
func pop(head *Student) (int, bool) { //从栈中弹出最后一个元素
	if head == nil { //判断是否为空链表
		return 0, false
	}
	p := head //设置临时指针,指向头节点
	for {
		if p.Next.Next == nil { //判断下一个节点的next值是否为空,如果为空
			num := p.Next.idnum //将下一个节点的idnum进行临时保存
			p.Next = nil        //改变此节点的next地址,改为空,在之后访问到此处,会停止循环
			return num, true    //取到最后的元素,则会返回元素,返回true
		}
		p = p.Next //如果不是最后一个节点,则进行下一个节点的遍历
	}
}
func main() {
	stu := Student{idnum: 0, Next: nil} //声明一结构体的对象,next值为nil
	stu.base = &stu
	stu.top = &stu
	push(&stu, []int{1, 2, 3, 4, 5, 6, 7}, stu.top) //将数组的元素放入链表中
	fmt.Println(pop(&stu))                          //使用pop弹出最后一个元素,并返回最后一个元素值
	p := &stu
	for { //遍历链表,将链表中的值输出,以确定是否将数组中的元素放入链表,并确定是否将链表的最后一个值弹出
		if p != nil {
			fmt.Println(p.idnum)
			p = p.Next
		} else {
			break
		}
	}
}