1. 程式人生 > 實用技巧 >go 實現棧

go 實現棧

使用切片的方式實現簡單的棧

package main

import "fmt"
type Element interface{}//可存入任何型別


type Stack struct {
	list []Element
}

//初始化棧
func NewStack()*Stack {
	return &Stack{
		list: make([]Element,0),
	}
}

func (s*Stack)Len()int  {
	return len(s.list)
}

//判斷棧是否空
func (s*Stack)IsEmpty()bool  {
	if len(s.list)==0{
		return true
	}else{
		return false
	}
}

//入棧
func (s *Stack)Push(x interface{})  {
	s.list = append(s.list,x)
}

//連續傳入
func (s *Stack)PushList(x []Element)  {
	s.list = append(s.list,x...)
}

//出棧
func (s*Stack)Pop() Element{
	if len(s.list)<=0{
		fmt.Println("Stack is Empty")
		return nil
	}else {
		ret:=s.list[len(s.list)-1]
		s.list = s.list[:len(s.list)-1]
		return ret
	}
}

//返回棧頂元素,空棧返nil
func (s*Stack)Top() Element {
	if s.IsEmpty()==true{
		fmt.Println("Stack is Empty")
		return nil
	}else{
		return s.list[len(s.list)-1]
	}
}

//清空棧
func (s*Stack)Clear()bool{
	if len(s.list)==0{
		return true
	}
	for i:=0;i<s.Len();i++{
		s.list[i]=nil
	}
	s.list = make([]Element,0)
	return true
}

//列印測試
func (s *Stack)Show()  {
	len:=len(s.list)
	for i:=0;i!=len;i++{
		fmt.Println(s.Pop())
	}
}

func main()  {
	stack:=NewStack()
	stack.Push(1)
	stack.Push(2)
	stack.Push(3)
	stack.Pop()
	stack.PushList([]Element{4,5,6,7})//list方式放入1
	fmt.Println("Stack len is ",stack.Len())
	fmt.Println("Top is",stack.Top())//棧頂元素
	stack.Show()//show之後資料已經清空
}