GO语言栈（链表实现）

Love The Way You Lie 2022-12-21 00:46 218阅读 0赞

栈，先进后出的规则

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
    		}
    	}
    }