栈,先进后出的规则
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
}
}
}