go stack 数据结构

最新推荐文章于 2024-12-03 19:41:01 发布

原创最新推荐文章于 2024-12-03 19:41:01 发布 · 296 阅读

0 ·

CC 4.0 BY-SA版权

go 专栏收录该内容

6 篇文章

订阅专栏

本文详细探讨了Go语言中的堆栈数据结构，包括其实现原理、操作（入栈、出栈）以及在并发环境下的使用。通过实例解析，展示了堆栈如何在内存管理中发挥关键作用，并讨论了其在解决编程问题，如回溯算法和函数调用跟踪等方面的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//stack.go

package stack 

type Stack struct{     //定义栈
	dataStore  []interface{}
	size int
}

func New() *Stack{   // 栈初始化
	s :=new(Stack)
	s.dataStore = make([]interface{},0)
	s.size = 0

	return s 
}

func (s *Stack) Size() int{   //返回栈的大小
	return s.size
}

func (s *Stack) IsEmpty() bool{  //判断栈是否为空
	return s.size == 0
}


func (s *Stack) Push(element interface{}){   //数据入栈操作
	s.dataStore = append(s.dataStore,element)
	s.size++
}


func (s *Stack) Pop() interface{} {   //数据出栈操作
	if(s.IsEmpty()){
		return nil
	}
	last :=s.dataStore[s.Size()-1]
	s.dataStore = s.dataStore[:s.Size()-1]
	s.size--
	return last

}


func (s *Stack) Peak() interface{} {  //取栈顶元素
	if(s.IsEmpty()){
		return nil
	}
	return s.dataStore[s.Size()-1]
}

func (s *Stack) Clear(){   //清空栈元素

	s.dataStore=s.dataStore[:1]
	s.size = 0 
}

//stack_test.go


package stack 

import "testing"


func TestStack(t *testing.T){
	stack :=New()
	if stack.Size()!=0 || !stack.IsEmpty(){
		t.Errorf("the size of stack is not equal 0")
	}

	stack.Push(1)
	stack.Push(3)

	if stack.Size()!=2 {
		t.Errorf("stack push failed!")
	}
	p :=stack.Pop()

	if v,ok :=p.(int);!ok||v!=3||stack.Size()!=1{
		t.Errorf("stack pop failed")
	}

	p2 :=stack.Peak()
	if v,ok :=p2.(int);!ok||v!=1||stack.Size()!=1{
		t.Log("stack peak error")
		t.Fail()
	}

	stack.Clear()
	if !stack.IsEmpty(){
		t.Errorf("stack clear error")
	}
}