golang算法—— 使用两个栈实现一个队列

前言

阅读本文，假定已经了解了基本数据结构概念。

队列: 先入先出。
栈: 先进后出。

分析

使用两个栈串联，可以实现先进先出。但是，得注意以下两点:

• 队列在入列时，stack2必须为空，stack1满员，保证顺序。
• 队列在出列时，stack1必须为空，stack2满员

实现

栈

type Stack struct {
	element []int
}

func NewStack(col int) *Stack {
	return &Stack{
		element: make([]int, 0, col),
	}
}

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

func (s *Stack) Push(elem int) {
	s.element = append(s.element, elem)
}

func (s *Stack) Pop() int {
	if len(s.element) == 0 {
		return 0
	}
	tmp := s.element[len(s.element)-1]
	s.element = s.element[:len(s.element)-1]
	return tmp
}

// 输出6 5
func TestStack(t *testing.T) {
	stack := NewStack(10)

	stack.Push(5)
	stack.Push(6)
	
	fmt.Println(stack.Pop())
	fmt.Println(stack.Pop())
}

队列

// 用两个stack，作一个队列
type Queue struct {
	stack1 *Stack
	stack2 *Stack
}

func NewQueue(col int) *Queue {
	return &Queue{
		stack1: NewStack(col),
		stack2: NewStack(col),
	}
}
func (q Queue) Push(elem int) {
	for q.stack2.Len() >0 {
		q.stack1.Push(q.stack2.Pop())
	}
	q.stack1.Push(elem)
}

func (q Queue) Pop() int {
	for q.stack1.Len() >0 {
		q.stack2.Push(q.stack1.Pop())
	}
	rs:=q.stack2.Pop()
	return rs
}

// 输出5 6 7 8
func TestQueue(t *testing.T) {
	queue := NewQueue(10)

	queue.Push(5)
	queue.Push(6)
	queue.Push(7)

	fmt.Println(queue.Pop())
	fmt.Println(queue.Pop())

	queue.Push(8)

	fmt.Println(queue.Pop())
	fmt.Println(queue.Pop())
}

实现仓库

https://github.com/fwhezfwhez/TestX/blob/0684f9df6f3a5c80f55c72f63d0843fbfd2aa432/test_practice/main_test.go#L73