Merge branch 'krahets:master' into master

This commit is contained in:
mgisr 2022-12-04 08:28:35 +08:00 committed by GitHub
commit bcc6aad18c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
48 changed files with 1068 additions and 501 deletions

View File

@ -58,7 +58,7 @@
## To-Dos ## To-Dos
- [x] [代码翻译](https://github.com/krahets/hello-algo/issues/15)JavaScript, TypeScript, C, C++, C#, ... 寻求大佬帮助) - [x] [代码翻译](https://github.com/krahets/hello-algo/issues/15)Java, C++, Python, Go, JavaScript 正在进行中,其他语言请求大佬挑大梁
- [ ] 数据结构:散列表、堆(优先队列)、图 - [ ] 数据结构:散列表、堆(优先队列)、图
- [ ] 算法:搜索与回溯、选择 / 堆排序、动态规划、贪心、分治 - [ ] 算法:搜索与回溯、选择 / 堆排序、动态规划、贪心、分治

View File

@ -21,7 +21,7 @@ int main() {
cout << "双向队列 deque = "; cout << "双向队列 deque = ";
PrintUtil::printDeque(deque); PrintUtil::printDeque(deque);
/* 访问队首元素 */ /* 访问元素 */
int front = deque.front(); int front = deque.front();
cout << "队首元素 front = " << front << endl; cout << "队首元素 front = " << front << endl;
int back = deque.back(); int back = deque.back();

View File

@ -5,6 +5,7 @@
package chapter_computational_complexity package chapter_computational_complexity
import ( import (
"fmt"
"testing" "testing"
) )
@ -16,8 +17,8 @@ func TestTwoSum(t *testing.T) {
// ====== Driver Code ====== // ====== Driver Code ======
// 方法一:暴力解法 // 方法一:暴力解法
res := twoSumBruteForce(nums, target) res := twoSumBruteForce(nums, target)
t.Log("方法一 res =", res) fmt.Println("方法一 res =", res)
// 方法二:哈希表 // 方法二:哈希表
res = twoSumHashTable(nums, target) res = twoSumHashTable(nums, target)
t.Log("方法二 res =", res) fmt.Println("方法二 res =", res)
} }

View File

@ -5,6 +5,7 @@
package chapter_searching package chapter_searching
import ( import (
"fmt"
"testing" "testing"
. "github.com/krahets/hello-algo/pkg" . "github.com/krahets/hello-algo/pkg"
@ -16,10 +17,10 @@ func TestLinearSearch(t *testing.T) {
// 在数组中执行线性查找 // 在数组中执行线性查找
index := linerSearchArray(nums, target) index := linerSearchArray(nums, target)
t.Log("目标元素 3 的索引 =", index) fmt.Println("目标元素 3 的索引 =", index)
// 在链表中执行线性查找 // 在链表中执行线性查找
head := ArrayToLinkedList(nums) head := ArrayToLinkedList(nums)
node := linerSearchLinkedList(head, target) node := linerSearchLinkedList(head, target)
t.Log("目标结点值 3 的对应结点对象为", node) fmt.Println("目标结点值 3 的对应结点对象为", node)
} }

View File

@ -4,50 +4,45 @@
package chapter_stack_and_queue package chapter_stack_and_queue
import ( /* 基于环形数组实现的队列 */
"fmt"
"strings"
)
// ArrayQueue 基于环形数组实现的队列, 不支持扩容
type ArrayQueue struct { type ArrayQueue struct {
data []any // 用于存储队列元素的数组 data []int // 用于存储队列元素的数组
capacity int // 队列容量(即最多容量的元素个数) capacity int // 队列容量(即最多容量的元素个数)
head int // 头指针,指向队首 front int // 头指针,指向队首
tail int // 尾指针,指向队尾 + 1 rear int // 尾指针,指向队尾 + 1
} }
// NewArrayQueue 基于环形数组实现的队列 // NewArrayQueue 基于环形数组实现的队列
func NewArrayQueue(capacity int) *ArrayQueue { func NewArrayQueue(capacity int) *ArrayQueue {
return &ArrayQueue{ return &ArrayQueue{
data: make([]any, capacity), data: make([]int, capacity),
capacity: capacity, capacity: capacity,
head: 0, front: 0,
tail: 0, rear: 0,
} }
} }
// Size 获取队列的长度 // Size 获取队列的长度
func (q *ArrayQueue) Size() int { func (q *ArrayQueue) Size() int {
size := (q.capacity + q.tail - q.head) % q.capacity size := (q.capacity + q.rear - q.front) % q.capacity
return size return size
} }
// IsEmpty 判断队列是否为空 // IsEmpty 判断队列是否为空
func (q *ArrayQueue) IsEmpty() bool { func (q *ArrayQueue) IsEmpty() bool {
return q.tail-q.head == 0 return q.rear-q.front == 0
} }
// Offer 入队 // Offer 入队
func (q *ArrayQueue) Offer(v any) { func (q *ArrayQueue) Offer(v int) {
// 当 tail == capacity 表示队列已满 // 当 rear == capacity 表示队列已满
if q.Size() == q.capacity { if q.Size() == q.capacity {
return return
} }
// 尾结点后添加 // 尾结点后添加
q.data[q.tail] = v q.data[q.rear] = v
// 尾指针向后移动一位,越过尾部后返回到数组头部 // 尾指针向后移动一位,越过尾部后返回到数组头部
q.tail = (q.tail + 1) % q.capacity q.rear = (q.rear + 1) % q.capacity
} }
// Poll 出队 // Poll 出队
@ -55,9 +50,9 @@ func (q *ArrayQueue) Poll() any {
if q.IsEmpty() { if q.IsEmpty() {
return nil return nil
} }
v := q.data[q.head] v := q.data[q.front]
// 队头指针向后移动一位,若越过尾部则返回到数组头部 // 队头指针向后移动一位,若越过尾部则返回到数组头部
q.head = (q.head + 1) % q.capacity q.front = (q.front + 1) % q.capacity
return v return v
} }
@ -66,27 +61,11 @@ func (q *ArrayQueue) Peek() any {
if q.IsEmpty() { if q.IsEmpty() {
return nil return nil
} }
v := q.data[q.head] v := q.data[q.front]
return v return v
} }
func (q *ArrayQueue) Print() { // 获取 Slice 用于打印
fmt.Println(q.toString()) func (s *ArrayQueue) toSlice() []int {
} return s.data[s.front:s.rear]
// toString 通过字符串的方式输出
func (q *ArrayQueue) toString() string {
// 为空时
if q.IsEmpty() {
return "empty items"
}
var builder strings.Builder
size := q.Size()
str := fmt.Sprintf("%+v", q.data[q.head])
for i := 1; i < size; i++ {
builder.WriteString(str + " -> ")
str = fmt.Sprintf("%+v", q.data[(i+q.head)%q.capacity])
}
builder.WriteString(str)
return builder.String()
} }

View File

@ -4,13 +4,7 @@
package chapter_stack_and_queue package chapter_stack_and_queue
import ( /* 基于数组实现的栈 */
"fmt"
"strconv"
"strings"
)
// ArrayStack 基于数组实现的栈
type ArrayStack struct { type ArrayStack struct {
data []int // 数据 data []int // 数据
} }
@ -38,6 +32,7 @@ func (s *ArrayStack) Push(v int) {
s.data = append(s.data, v) s.data = append(s.data, v)
} }
// Pop 出栈
func (s *ArrayStack) Pop() any { func (s *ArrayStack) Pop() any {
// 弹出栈前,先判断是否为空 // 弹出栈前,先判断是否为空
if s.IsEmpty() { if s.IsEmpty() {
@ -48,6 +43,7 @@ func (s *ArrayStack) Pop() any {
return val return val
} }
// Peek 获取栈顶元素
func (s *ArrayStack) Peek() any { func (s *ArrayStack) Peek() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
@ -56,18 +52,7 @@ func (s *ArrayStack) Peek() any {
return val return val
} }
func (s *ArrayStack) Print() { // 获取 Slice 用于打印
fmt.Println(s.toString()) func (s *ArrayStack) toSlice() []int {
} return s.data
func (s *ArrayStack) toString() string {
var builder strings.Builder
if s.IsEmpty() {
return "empty stack"
}
for i := len(s.data) - 1; i > 0; i-- {
builder.WriteString(strconv.Itoa(s.data[i]) + " -> ")
}
builder.WriteString(strconv.Itoa(s.data[0]))
return builder.String()
} }

View File

@ -1,26 +0,0 @@
// File: deque.go
// Created Time: 2022-11-29
// Author: Reanon (793584285@qq.com)
package chapter_stack_and_queue
type Deque interface {
// OfferFirst 元素入队
OfferFirst(num any)
// OfferLast 元素入队
OfferLast(num any)
// PeekFirst 访问首元素
PeekFirst() any
// PeekLast 访问尾元素
PeekLast() any
// PollFirst 元素出队
PollFirst() any
// PollLast 元素出队
PollLast() any
// Size 获取队列长度
Size() int
// IsEmpty 队列是否为空
IsEmpty() bool
// toString 队列输出为字符串
toString() string
}

View File

@ -4,12 +4,54 @@
package chapter_stack_and_queue package chapter_stack_and_queue
import "testing" import (
"container/list"
"fmt"
"testing"
. "github.com/krahets/hello-algo/pkg"
)
func TestDeque(t *testing.T) {
/* 初始化双向队列 */
// 在 Go 中,将 list 作为双向队列使用
deque := list.New()
/* 元素入队 */
deque.PushBack(2)
deque.PushBack(5)
deque.PushBack(4)
deque.PushFront(3)
deque.PushFront(1)
fmt.Print("双向队列 deque = ")
PrintList(deque)
/* 访问元素 */
front := deque.Front()
fmt.Println("队首元素 front =", front.Value)
rear := deque.Back()
fmt.Println("队尾元素 rear =", rear.Value)
/* 元素出队 */
deque.Remove(front)
fmt.Print("队首出队元素 front = ", front.Value, ",队首出队后 deque = ")
PrintList(deque)
deque.Remove(rear)
fmt.Print("队尾出队元素 rear = ", rear.Value, ",队尾出队后 deque = ")
PrintList(deque)
/* 获取双向队列的长度 */
size := deque.Len()
fmt.Println("双向队列长度 size =", size)
/* 判断双向队列是否为空 */
isEmpty := deque.Len() == 0
fmt.Println("双向队列是否为空 =", isEmpty)
}
func TestLinkedListDeque(t *testing.T) { func TestLinkedListDeque(t *testing.T) {
// 初始化队列 // 初始化队列
var deque Deque deque := NewLinkedListDeque()
deque = NewLinkedListDeque()
// 元素入队 // 元素入队
deque.OfferLast(2) deque.OfferLast(2)
@ -17,27 +59,30 @@ func TestLinkedListDeque(t *testing.T) {
deque.OfferLast(4) deque.OfferLast(4)
deque.OfferFirst(3) deque.OfferFirst(3)
deque.OfferFirst(1) deque.OfferFirst(1)
t.Log("队列 deque = ", deque.toString()) fmt.Print("队列 deque = ")
PrintList(deque.toList())
// 访问队首元素 // 访问队首元素
peekFirst := deque.PeekFirst() front := deque.PeekFirst()
t.Log("队首元素 peek = ", peekFirst) fmt.Println("队首元素 front =", front)
peekLast := deque.PeekLast() rear := deque.PeekLast()
t.Log("队尾元素 peek = ", peekLast) fmt.Println("队尾元素 rear =", rear)
// 元素出队 // 元素出队
pollFirst := deque.PollFirst() pollFirst := deque.PollFirst()
t.Log("队首出队元素 pollFirst = ", pollFirst, ",队首出队后 deque = ", deque.toString()) fmt.Print("队首出队元素 pollFirst = ", pollFirst, ",队首出队后 deque = ")
PrintList(deque.toList())
pollLast := deque.PollLast() pollLast := deque.PollLast()
t.Log("队尾出队元素 pollLast = ", pollLast, ",队尾出队后 deque = ", deque.toString()) fmt.Print("队尾出队元素 pollLast = ", pollLast, ",队尾出队后 deque = ")
PrintList(deque.toList())
// 获取队的长度 // 获取队的长度
size := deque.Size() size := deque.Size()
t.Log("队的长度 size = ", size) fmt.Println("队的长度 size =", size)
// 判断是否为空 // 判断是否为空
isEmpty := deque.IsEmpty() isEmpty := deque.IsEmpty()
t.Log("队是否为空 = ", isEmpty) fmt.Println("队是否为空 =", isEmpty)
} }
// BenchmarkArrayQueue 67.92 ns/op in Mac M1 Pro // BenchmarkArrayQueue 67.92 ns/op in Mac M1 Pro

View File

@ -6,97 +6,79 @@ package chapter_stack_and_queue
import ( import (
"container/list" "container/list"
"fmt"
"strings"
) )
// LinkedListDeque 基于链表实现的双端队列, 使用内置包 list 来实现栈 // LinkedListDeque 基于链表实现的双端队列, 使用内置包 list 来实现栈
type LinkedListDeque struct { type LinkedListDeque struct {
list *list.List data *list.List
} }
// NewLinkedListDeque 初始化双端队列 // NewLinkedListDeque 初始化双端队列
func NewLinkedListDeque() *LinkedListDeque { func NewLinkedListDeque() *LinkedListDeque {
return &LinkedListDeque{ return &LinkedListDeque{
list: list.New(), data: list.New(),
} }
} }
// OfferFirst 元素入队 // OfferFirst 队首元素入队
func (s *LinkedListDeque) OfferFirst(value any) { func (s *LinkedListDeque) OfferFirst(value any) {
s.list.PushFront(value) s.data.PushFront(value)
} }
// OfferLast 元素入队 // OfferLast 队尾元素入队
func (s *LinkedListDeque) OfferLast(value any) { func (s *LinkedListDeque) OfferLast(value any) {
s.list.PushBack(value) s.data.PushBack(value)
} }
// PollFirst 元素出队 // PollFirst 队首元素出队
func (s *LinkedListDeque) PollFirst() any { func (s *LinkedListDeque) PollFirst() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Front() e := s.data.Front()
s.list.Remove(e) s.data.Remove(e)
return e.Value return e.Value
} }
// PollLast 元素出队 // PollLast 队尾元素出队
func (s *LinkedListDeque) PollLast() any { func (s *LinkedListDeque) PollLast() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Back() e := s.data.Back()
s.list.Remove(e) s.data.Remove(e)
return e.Value return e.Value
} }
// PeekFirst 访问首元素 // PeekFirst 访问首元素
func (s *LinkedListDeque) PeekFirst() any { func (s *LinkedListDeque) PeekFirst() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Front() e := s.data.Front()
return e.Value return e.Value
} }
// PeekLast 访问尾元素 // PeekLast 访问尾元素
func (s *LinkedListDeque) PeekLast() any { func (s *LinkedListDeque) PeekLast() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Back() e := s.data.Back()
return e.Value return e.Value
} }
// Size 获取队列的长度 // Size 获取队列的长度
func (s *LinkedListDeque) Size() int { func (s *LinkedListDeque) Size() int {
return s.list.Len() return s.data.Len()
} }
// IsEmpty 判断队列是否为空 // IsEmpty 判断队列是否为空
func (s *LinkedListDeque) IsEmpty() bool { func (s *LinkedListDeque) IsEmpty() bool {
return s.list.Len() == 0 return s.data.Len() == 0
} }
func (s *LinkedListDeque) Print() { // 获取 List 用于打印
fmt.Println(s.toString()) func (s *LinkedListDeque) toList() *list.List {
} return s.data
func (s *LinkedListDeque) toString() string {
var builder strings.Builder
if s.IsEmpty() {
fmt.Println("empty stack")
}
e := s.list.Front()
// 强转为 string, 会影响效率
str := fmt.Sprintf("%v", e.Value)
for e.Next() != nil {
builder.WriteString(str + " -> ")
e = e.Next()
str = fmt.Sprintf("%v", e.Value)
}
builder.WriteString(str)
return builder.String()
} }

View File

@ -6,25 +6,24 @@ package chapter_stack_and_queue
import ( import (
"container/list" "container/list"
"fmt"
"strings"
) )
// LinkedListQueue 基于链表实现的队列, 使用内置包 list 来实现栈 /* 基于链表实现的队列 */
type LinkedListQueue struct { type LinkedListQueue struct {
list *list.List // 使用内置包 list 来实现队列
data *list.List
} }
// NewLinkedListQueue 初始化链表 // NewLinkedListQueue 初始化链表
func NewLinkedListQueue() *LinkedListQueue { func NewLinkedListQueue() *LinkedListQueue {
return &LinkedListQueue{ return &LinkedListQueue{
list: list.New(), data: list.New(),
} }
} }
// Offer 入队 // Offer 入队
func (s *LinkedListQueue) Offer(value any) { func (s *LinkedListQueue) Offer(value any) {
s.list.PushBack(value) s.data.PushBack(value)
} }
// Poll 出队 // Poll 出队
@ -32,8 +31,8 @@ func (s *LinkedListQueue) Poll() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Front() e := s.data.Front()
s.list.Remove(e) s.data.Remove(e)
return e.Value return e.Value
} }
@ -42,37 +41,21 @@ func (s *LinkedListQueue) Peek() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Front() e := s.data.Front()
return e.Value return e.Value
} }
// Size 获取队列的长度 // Size 获取队列的长度
func (s *LinkedListQueue) Size() int { func (s *LinkedListQueue) Size() int {
return s.list.Len() return s.data.Len()
} }
// IsEmpty 判断队列是否为空 // IsEmpty 判断队列是否为空
func (s *LinkedListQueue) IsEmpty() bool { func (s *LinkedListQueue) IsEmpty() bool {
return s.list.Len() == 0 return s.data.Len() == 0
} }
func (s *LinkedListQueue) Print() { // 获取 List 用于打印
fmt.Println(s.toString()) func (s *LinkedListQueue) toList() *list.List {
} return s.data
func (s *LinkedListQueue) toString() string {
var builder strings.Builder
if s.IsEmpty() {
fmt.Println("empty stack")
}
e := s.list.Front()
// 强转为 string, 会影响效率
str := fmt.Sprintf("%v", e.Value)
for e.Next() != nil {
builder.WriteString(str + " -> ")
e = e.Next()
str = fmt.Sprintf("%v", e.Value)
}
builder.WriteString(str)
return builder.String()
} }

View File

@ -6,25 +6,24 @@ package chapter_stack_and_queue
import ( import (
"container/list" "container/list"
"fmt"
"strings"
) )
// LinkedListStack 基于链表实现的栈, 使用内置包 list 来实现栈 /* 基于链表实现的栈 */
type LinkedListStack struct { type LinkedListStack struct {
list *list.List // 使用内置包 list 来实现栈
data *list.List
} }
// NewLinkedListStack 初始化链表 // NewLinkedListStack 初始化链表
func NewLinkedListStack() *LinkedListStack { func NewLinkedListStack() *LinkedListStack {
return &LinkedListStack{ return &LinkedListStack{
list: list.New(), data: list.New(),
} }
} }
// Push 入栈 // Push 入栈
func (s *LinkedListStack) Push(value int) { func (s *LinkedListStack) Push(value int) {
s.list.PushBack(value) s.data.PushBack(value)
} }
// Pop 出栈 // Pop 出栈
@ -32,8 +31,8 @@ func (s *LinkedListStack) Pop() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Back() e := s.data.Back()
s.list.Remove(e) s.data.Remove(e)
return e.Value return e.Value
} }
@ -42,37 +41,21 @@ func (s *LinkedListStack) Peek() any {
if s.IsEmpty() { if s.IsEmpty() {
return nil return nil
} }
e := s.list.Back() e := s.data.Back()
return e.Value return e.Value
} }
// Size 获取栈的长度 // Size 获取栈的长度
func (s *LinkedListStack) Size() int { func (s *LinkedListStack) Size() int {
return s.list.Len() return s.data.Len()
} }
// IsEmpty 判断栈是否为空 // IsEmpty 判断栈是否为空
func (s *LinkedListStack) IsEmpty() bool { func (s *LinkedListStack) IsEmpty() bool {
return s.list.Len() == 0 return s.data.Len() == 0
} }
func (s *LinkedListStack) Print() { // 获取 List 用于打印
fmt.Println(s.toString()) func (s *LinkedListStack) toList() *list.List {
} return s.data
func (s *LinkedListStack) toString() string {
var builder strings.Builder
if s.IsEmpty() {
fmt.Println("empty stack")
}
e := s.list.Back()
// 强转为 string, 会影响效率
str := fmt.Sprintf("%v", e.Value)
for e.Prev() != nil {
builder.WriteString(str + " -> ")
e = e.Prev()
str = fmt.Sprintf("%v", e.Value)
}
builder.WriteString(str)
return builder.String()
} }

View File

@ -1,20 +0,0 @@
// File: queue.go
// Created Time: 2022-11-29
// Author: Reanon (793584285@qq.com)
package chapter_stack_and_queue
type Queue interface {
// Offer 元素入队
Offer(num any)
// Peek 访问首元素
Peek() any
// Poll 元素出队
Poll() any
// Size 获取队列长度
Size() int
// IsEmpty 队列是否为空
IsEmpty() bool
// toString 队列输出为字符串
toString() string
}

View File

@ -4,67 +4,108 @@
package chapter_stack_and_queue package chapter_stack_and_queue
import "testing" import (
"container/list"
"fmt"
"testing"
. "github.com/krahets/hello-algo/pkg"
)
func TestQueue(t *testing.T) {
/* 初始化队列 */
// 在 Go 中,将 list 作为队列来使用
queue := list.New()
/* 元素入队 */
queue.PushBack(1)
queue.PushBack(3)
queue.PushBack(2)
queue.PushBack(5)
queue.PushBack(4)
fmt.Print("队列 queue = ")
PrintList(queue)
/* 访问队首元素 */
peek := queue.Front()
fmt.Println("队首元素 peek =", peek.Value)
/* 元素出队 */
poll := queue.Front()
queue.Remove(poll)
fmt.Print("出队元素 poll = ", poll.Value, ",出队后 queue = ")
PrintList(queue)
/* 获取队列的长度 */
size := queue.Len()
fmt.Println("队列长度 size =", size)
/* 判断队列是否为空 */
isEmpty := queue.Len() == 0
fmt.Println("队列是否为空 =", isEmpty)
}
func TestArrayQueue(t *testing.T) { func TestArrayQueue(t *testing.T) {
// 初始化队列,使用队列的通用接口 // 初始化队列,使用队列的通用接口
var queue Queue
capacity := 10 capacity := 10
queue = NewArrayQueue(capacity) queue := NewArrayQueue(capacity)
// 元素入队 // 元素入队
queue.Offer(1) queue.Offer(1)
queue.Offer(2)
queue.Offer(3) queue.Offer(3)
queue.Offer(4) queue.Offer(2)
queue.Offer(5) queue.Offer(5)
t.Log("队列 queue = ", queue.toString()) queue.Offer(4)
fmt.Print("队列 queue = ")
PrintSlice(queue.toSlice())
// 访问队首元素 // 访问队首元素
peek := queue.Peek() peek := queue.Peek()
t.Log("队首元素 peek = ", peek) fmt.Println("队首元素 peek =", peek)
// 元素出队 // 元素出队
pop := queue.Poll() poll := queue.Poll()
t.Log("出队元素 pop = ", pop, ", 出队后 queue =", queue.toString()) fmt.Print("出队元素 poll = ", poll, ", 出队后 queue = ")
PrintSlice(queue.toSlice())
// 获取队的长度 // 获取队的长度
size := queue.Size() size := queue.Size()
t.Log("队的长度 size = ", size) fmt.Println("队的长度 size =", size)
// 判断是否为空 // 判断是否为空
isEmpty := queue.IsEmpty() isEmpty := queue.IsEmpty()
t.Log("队是否为空 = ", isEmpty) fmt.Println("队是否为空 =", isEmpty)
} }
func TestLinkedListQueue(t *testing.T) { func TestLinkedListQueue(t *testing.T) {
// 初始化队 // 初始化队
var queue Queue queue := NewLinkedListQueue()
queue = NewLinkedListQueue()
// 元素入队 // 元素入队
queue.Offer(1) queue.Offer(1)
queue.Offer(2)
queue.Offer(3) queue.Offer(3)
queue.Offer(4) queue.Offer(2)
queue.Offer(5) queue.Offer(5)
t.Log("队列 queue = ", queue.toString()) queue.Offer(4)
fmt.Print("队列 queue = ")
PrintList(queue.toList())
// 访问队首元素 // 访问队首元素
peek := queue.Peek() peek := queue.Peek()
t.Log("队首元素 peek = ", peek) fmt.Println("队首元素 peek =", peek)
// 元素出队 // 元素出队
pop := queue.Poll() poll := queue.Poll()
t.Log("出队元素 pop = ", pop, ", 出队后 queue =", queue.toString()) fmt.Print("出队元素 poll = ", poll, ", 出队后 queue = ")
PrintList(queue.toList())
// 获取队的长度 // 获取队的长度
size := queue.Size() size := queue.Size()
t.Log("队的长度 size = ", size) fmt.Println("队的长度 size =", size)
// 判断是否为空 // 判断是否为空
isEmpty := queue.IsEmpty() isEmpty := queue.IsEmpty()
t.Log("队是否为空 = ", isEmpty) fmt.Println("队是否为空 =", isEmpty)
} }
// BenchmarkArrayQueue 8 ns/op in Mac M1 Pro // BenchmarkArrayQueue 8 ns/op in Mac M1 Pro

View File

@ -1,20 +0,0 @@
// File: stack.go
// Created Time: 2022-11-26
// Author: Reanon (793584285@qq.com)
package chapter_stack_and_queue
type Stack interface {
// Push 元素入栈
Push(num int)
// Peek 访问栈顶元素
Peek() any
// Pop 元素出栈
Pop() any
// Size 栈的长度
Size() int
// IsEmpty 栈是否为空
IsEmpty() bool
// toString 栈输出为字符串
toString() string
}

View File

@ -4,65 +4,105 @@
package chapter_stack_and_queue package chapter_stack_and_queue
import "testing" import (
"fmt"
"testing"
. "github.com/krahets/hello-algo/pkg"
)
func TestStack(t *testing.T) {
/* 初始化栈 */
// 在 Go 中,推荐将 Slice 当作栈来使用
var stack []int
/* 元素入栈 */
stack = append(stack, 1)
stack = append(stack, 3)
stack = append(stack, 2)
stack = append(stack, 5)
stack = append(stack, 4)
fmt.Print("栈 = ")
PrintSlice(stack)
/* 访问栈顶元素 */
peek := stack[len(stack)-1]
fmt.Println("栈顶元素 peek =", peek)
/* 元素出栈 */
pop := stack[len(stack)-1]
stack = stack[:len(stack)-1]
fmt.Print("出栈元素 pop = ", pop, ",出栈后 stack = ")
PrintSlice(stack)
/* 获取栈的长度 */
size := len(stack)
fmt.Println("栈的长度 size =", size)
/* 判断是否为空 */
isEmpty := len(stack) == 0
fmt.Println("栈是否为空 =", isEmpty)
}
func TestArrayStack(t *testing.T) { func TestArrayStack(t *testing.T) {
// 初始化栈, 使用接口承接 // 初始化栈, 使用接口承接
var stack Stack stack := NewArrayStack()
stack = NewArrayStack()
// 元素入栈 // 元素入栈
stack.Push(1) stack.Push(1)
stack.Push(2)
stack.Push(3) stack.Push(3)
stack.Push(4) stack.Push(2)
stack.Push(5) stack.Push(5)
t.Log("栈 stack = ", stack.toString()) stack.Push(4)
fmt.Print("栈 stack = ")
PrintSlice(stack.toSlice())
// 访问栈顶元素 // 访问栈顶元素
peek := stack.Peek() peek := stack.Peek()
t.Log("栈顶元素 peek = ", peek) fmt.Println("栈顶元素 peek =", peek)
// 元素出栈 // 元素出栈
pop := stack.Pop() pop := stack.Pop()
t.Log("出栈元素 pop = ", pop, ", 出栈后 stack =", stack.toString()) fmt.Print("出栈元素 pop = ", pop, ", 出栈后 stack = ")
PrintSlice(stack.toSlice())
// 获取栈的长度 // 获取栈的长度
size := stack.Size() size := stack.Size()
t.Log("栈的长度 size = ", size) fmt.Println("栈的长度 size =", size)
// 判断是否为空 // 判断是否为空
isEmpty := stack.IsEmpty() isEmpty := stack.IsEmpty()
t.Log("栈是否为空 = ", isEmpty) fmt.Println("栈是否为空 =", isEmpty)
} }
func TestLinkedListStack(t *testing.T) { func TestLinkedListStack(t *testing.T) {
// 初始化栈 // 初始化栈
var stack Stack stack := NewLinkedListStack()
stack = NewLinkedListStack()
// 元素入栈 // 元素入栈
stack.Push(1) stack.Push(1)
stack.Push(2)
stack.Push(3) stack.Push(3)
stack.Push(4) stack.Push(2)
stack.Push(5) stack.Push(5)
t.Log("栈 stack = ", stack.toString()) stack.Push(4)
fmt.Print("栈 stack = ")
PrintList(stack.toList())
// 访问栈顶元素 // 访问栈顶元素
peek := stack.Peek() peek := stack.Peek()
t.Log("栈顶元素 peek = ", peek) fmt.Println("栈顶元素 peek =", peek)
// 元素出栈 // 元素出栈
pop := stack.Pop() pop := stack.Pop()
t.Log("出栈元素 pop = ", pop, ", 出栈后 stack =", stack.toString()) fmt.Print("出栈元素 pop = ", pop, ", 出栈后 stack = ")
PrintList(stack.toList())
// 获取栈的长度 // 获取栈的长度
size := stack.Size() size := stack.Size()
t.Log("栈的长度 size = ", size) fmt.Println("栈的长度 size =", size)
// 判断是否为空 // 判断是否为空
isEmpty := stack.IsEmpty() isEmpty := stack.IsEmpty()
t.Log("栈是否为空 = ", isEmpty) fmt.Println("栈是否为空 =", isEmpty)
} }
// BenchmarkArrayStack 8 ns/op in Mac M1 Pro // BenchmarkArrayStack 8 ns/op in Mac M1 Pro

View File

@ -53,7 +53,7 @@ func (bst *BinarySearchTree) GetInorderNext(node *TreeNode) *TreeNode {
return node return node
} }
// Search node of binary search tree /* 查找结点 */
func (bst *BinarySearchTree) Search(num int) *TreeNode { func (bst *BinarySearchTree) Search(num int) *TreeNode {
node := bst.root node := bst.root
// 循环查找,越过叶结点后跳出 // 循环查找,越过叶结点后跳出
@ -73,7 +73,7 @@ func (bst *BinarySearchTree) Search(num int) *TreeNode {
return node return node
} }
// Insert node of binary search tree /* 插入结点 */
func (bst *BinarySearchTree) Insert(num int) *TreeNode { func (bst *BinarySearchTree) Insert(num int) *TreeNode {
cur := bst.root cur := bst.root
// 若树为空,直接提前返回 // 若树为空,直接提前返回
@ -104,7 +104,7 @@ func (bst *BinarySearchTree) Insert(num int) *TreeNode {
return cur return cur
} }
// Remove node of binary search tree /* 删除结点 */
func (bst *BinarySearchTree) Remove(num int) *TreeNode { func (bst *BinarySearchTree) Remove(num int) *TreeNode {
cur := bst.root cur := bst.root
// 若树为空,直接提前返回 // 若树为空,直接提前返回
@ -119,8 +119,8 @@ func (bst *BinarySearchTree) Remove(num int) *TreeNode {
break break
} }
prev = cur prev = cur
// 待删除结点在右子树中
if cur.Val < num { if cur.Val < num {
// 待删除结点在右子树中
cur = cur.Right cur = cur.Right
} else { } else {
// 待删除结点在左子树中 // 待删除结点在左子树中
@ -146,8 +146,8 @@ func (bst *BinarySearchTree) Remove(num int) *TreeNode {
} else { } else {
prev.Right = child prev.Right = child
} }
// 子结点数为 2
} else { // 子结点数为 2 } else {
// 获取中序遍历中待删除结点 cur 的下一个结点 // 获取中序遍历中待删除结点 cur 的下一个结点
next := bst.GetInorderNext(cur) next := bst.GetInorderNext(cur)
temp := next.Val temp := next.Val
@ -156,7 +156,6 @@ func (bst *BinarySearchTree) Remove(num int) *TreeNode {
// 将 next 的值复制给 cur // 将 next 的值复制给 cur
cur.Val = temp cur.Val = temp
} }
// TODO: add error handler, don't return node
return cur return cur
} }

View File

@ -4,38 +4,41 @@
package chapter_tree package chapter_tree
import "testing" import (
"fmt"
"testing"
)
func TestBinarySearchTree(t *testing.T) { func TestBinarySearchTree(t *testing.T) {
nums := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15} nums := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15}
bst := NewBinarySearchTree(nums) bst := NewBinarySearchTree(nums)
t.Log("初始化的二叉树为:") fmt.Println("初始化的二叉树为:")
bst.Print() bst.Print()
// 获取根结点 // 获取根结点
node := bst.GetRoot() node := bst.GetRoot()
t.Log("二叉树的根结点为:", node.Val) fmt.Println("二叉树的根结点为:", node.Val)
// 获取最小的结点 // 获取最小的结点
node = bst.GetMin(bst.GetRoot()) node = bst.GetMin(bst.GetRoot())
t.Log("二叉树的最小结点为:", node.Val) fmt.Println("二叉树的最小结点为:", node.Val)
// 查找结点 // 查找结点
node = bst.Search(5) node = bst.Search(5)
t.Log("查找到的结点对象为", node, ",结点值 =", node.Val) fmt.Println("查找到的结点对象为", node, ",结点值 =", node.Val)
// 插入结点 // 插入结点
node = bst.Insert(16) node = bst.Insert(16)
t.Log("插入结点后 16 的二叉树为:") fmt.Println("插入结点后 16 的二叉树为:")
bst.Print() bst.Print()
// 删除结点 // 删除结点
bst.Remove(1) bst.Remove(1)
t.Log("删除结点 1 后的二叉树为:") fmt.Println("删除结点 1 后的二叉树为:")
bst.Print() bst.Print()
bst.Remove(2) bst.Remove(2)
t.Log("删除结点 2 后的二叉树为:") fmt.Println("删除结点 2 后的二叉树为:")
bst.Print() bst.Print()
bst.Remove(4) bst.Remove(4)
t.Log("删除结点 4 后的二叉树为:") fmt.Println("删除结点 4 后的二叉树为:")
bst.Print() bst.Print()
} }

View File

@ -1,23 +0,0 @@
// File: binary_tree.go
// Created Time: 2022-11-25
// Author: Reanon (793584285@qq.com)
package chapter_tree
import (
. "github.com/krahets/hello-algo/pkg"
)
type BinaryTree struct {
root *TreeNode
}
func NewBinaryTree(node *TreeNode) *BinaryTree {
return &BinaryTree{
root: node,
}
}
func (tree *BinaryTree) Print() {
PrintTree(tree.root)
}

View File

@ -10,11 +10,10 @@ import (
. "github.com/krahets/hello-algo/pkg" . "github.com/krahets/hello-algo/pkg"
) )
// levelOrder Breadth First Search /* 层序遍历 */
func levelOrder(root *TreeNode) []int { func levelOrder(root *TreeNode) []int {
// Let container.list as queue
queue := list.New()
// 初始化队列,加入根结点 // 初始化队列,加入根结点
queue := list.New()
queue.PushBack(root) queue.PushBack(root)
// 初始化一个切片,用于保存遍历序列 // 初始化一个切片,用于保存遍历序列
nums := make([]int, 0) nums := make([]int, 0)

View File

@ -5,6 +5,7 @@
package chapter_tree package chapter_tree
import ( import (
"fmt"
"testing" "testing"
. "github.com/krahets/hello-algo/pkg" . "github.com/krahets/hello-algo/pkg"
@ -14,10 +15,10 @@ func TestLevelOrder(t *testing.T) {
/* 初始化二叉树 */ /* 初始化二叉树 */
// 这里借助了一个从数组直接生成二叉树的函数 // 这里借助了一个从数组直接生成二叉树的函数
root := ArrayToTree([]int{1, 2, 3, 4, 5, 6, 7}) root := ArrayToTree([]int{1, 2, 3, 4, 5, 6, 7})
t.Log("初始化二叉树: ") fmt.Println("初始化二叉树: ")
PrintTree(root) PrintTree(root)
// 层序遍历 // 层序遍历
nums := levelOrder(root) nums := levelOrder(root)
t.Log("层序遍历的结点打印序列 =", nums) fmt.Println("层序遍历的结点打印序列 =", nums)
} }

View File

@ -8,56 +8,37 @@ import (
. "github.com/krahets/hello-algo/pkg" . "github.com/krahets/hello-algo/pkg"
) )
// preOrder 前序遍历 var nums []int
func preOrder(root *TreeNode) (nums []int) {
var preOrderHelper func(node *TreeNode) /* 前序遍历 */
// 匿名函数 func preOrder(node *TreeNode) {
preOrderHelper = func(node *TreeNode) { if node == nil {
if node == nil { return
return
}
// 访问优先级:根结点 -> 左子树 -> 右子树
nums = append(nums, node.Val)
preOrderHelper(node.Left)
preOrderHelper(node.Right)
} }
// 函数调用 // 访问优先级:根结点 -> 左子树 -> 右子树
preOrderHelper(root) nums = append(nums, node.Val)
return preOrder(node.Left)
preOrder(node.Right)
} }
// inOrder 中序遍历 /* 中序遍历 */
func inOrder(root *TreeNode) (nums []int) { func inOrder(node *TreeNode) {
var inOrderHelper func(node *TreeNode) if node == nil {
// 匿名函数 return
inOrderHelper = func(node *TreeNode) {
if node == nil {
return
}
// 访问优先级:左子树 -> 根结点 -> 右子树
inOrderHelper(node.Left)
nums = append(nums, node.Val)
inOrderHelper(node.Right)
} }
// 函数调用 // 访问优先级:左子树 -> 根结点 -> 右子树
inOrderHelper(root) inOrder(node.Left)
return nums = append(nums, node.Val)
inOrder(node.Right)
} }
// postOrder 后序遍历 /* 后序遍历 */
func postOrder(root *TreeNode) (nums []int) { func postOrder(node *TreeNode) {
var postOrderHelper func(node *TreeNode) if node == nil {
// 匿名函数 return
postOrderHelper = func(node *TreeNode) {
if node == nil {
return
}
// 访问优先级:左子树 -> 右子树 -> 根结点
postOrderHelper(node.Left)
postOrderHelper(node.Right)
nums = append(nums, node.Val)
} }
// 函数调用 // 访问优先级:左子树 -> 右子树 -> 根结点
postOrderHelper(root) postOrder(node.Left)
return postOrder(node.Right)
nums = append(nums, node.Val)
} }

View File

@ -5,6 +5,7 @@
package chapter_tree package chapter_tree
import ( import (
"fmt"
"testing" "testing"
. "github.com/krahets/hello-algo/pkg" . "github.com/krahets/hello-algo/pkg"
@ -14,18 +15,21 @@ func TestPreInPostOrderTraversal(t *testing.T) {
/* 初始化二叉树 */ /* 初始化二叉树 */
// 这里借助了一个从数组直接生成二叉树的函数 // 这里借助了一个从数组直接生成二叉树的函数
root := ArrayToTree([]int{1, 2, 3, 4, 5, 6, 7}) root := ArrayToTree([]int{1, 2, 3, 4, 5, 6, 7})
t.Log("初始化二叉树: ") fmt.Println("初始化二叉树: ")
PrintTree(root) PrintTree(root)
// 前序遍历 // 前序遍历
nums := preOrder(root) nums = nil
t.Log("前序遍历的结点打印序列 =", nums) preOrder(root)
fmt.Println("前序遍历的结点打印序列 =", nums)
// 中序遍历 // 中序遍历
nums = inOrder(root) nums = nil
t.Log("中序遍历的结点打印序列 =", nums) inOrder(root)
fmt.Println("中序遍历的结点打印序列 =", nums)
// 后序遍历 // 后序遍历
nums = postOrder(root) nums = nil
t.Log("后序遍历的结点打印序列 =", nums) postOrder(root)
fmt.Println("后序遍历的结点打印序列 =", nums)
} }

View File

@ -5,6 +5,7 @@
package chapter_tree package chapter_tree
import ( import (
"fmt"
"testing" "testing"
. "github.com/krahets/hello-algo/pkg" . "github.com/krahets/hello-algo/pkg"
@ -18,24 +19,23 @@ func TestBinaryTree(t *testing.T) {
n3 := NewTreeNode(3) n3 := NewTreeNode(3)
n4 := NewTreeNode(4) n4 := NewTreeNode(4)
n5 := NewTreeNode(5) n5 := NewTreeNode(5)
// 构建引用指向(即指针)
tree := NewBinaryTree(n1)
n1.Left = n2 n1.Left = n2
n1.Right = n3 n1.Right = n3
n2.Left = n4 n2.Left = n4
n2.Right = n5 n2.Right = n5
t.Log("初始化二叉树") fmt.Println("初始化二叉树")
tree.Print() PrintTree(n1)
/* 插入与删除结点 */ /* 插入与删除结点 */
// 插入结点
p := NewTreeNode(0) p := NewTreeNode(0)
n1.Left = p n1.Left = p
p.Left = n2 p.Left = n2
t.Log("插入结点 P 后") fmt.Println("插入结点 P 后")
tree.Print() PrintTree(n1)
// 删除结点 // 删除结点
n1.Left = n2 n1.Left = n2
t.Log("删除结点 P 后") fmt.Println("删除结点 P 后")
tree.Print() PrintTree(n1)
} }

View File

@ -4,12 +4,6 @@
package pkg package pkg
import (
"fmt"
"strconv"
"strings"
)
// ListNode Definition for a singly-linked list node // ListNode Definition for a singly-linked list node
type ListNode struct { type ListNode struct {
Next *ListNode Next *ListNode
@ -43,17 +37,3 @@ func GetListNode(node *ListNode, val int) *ListNode {
} }
return node return node
} }
// PrintLinkedList Print a linked list
func PrintLinkedList(node *ListNode) {
if node == nil {
return
}
var builder strings.Builder
for node.Next != nil {
builder.WriteString(strconv.Itoa(node.Val) + " -> ")
node = node.Next
}
builder.WriteString(strconv.Itoa(node.Val))
fmt.Println(builder.String())
}

View File

@ -4,7 +4,10 @@
package pkg package pkg
import "testing" import (
"fmt"
"testing"
)
func TestListNode(t *testing.T) { func TestListNode(t *testing.T) {
arr := []int{2, 3, 5, 6, 7} arr := []int{2, 3, 5, 6, 7}
@ -12,5 +15,5 @@ func TestListNode(t *testing.T) {
PrintLinkedList(head) PrintLinkedList(head)
node := GetListNode(head, 5) node := GetListNode(head, 5)
t.Log("Find node: ", node.Val) fmt.Println("Find node: ", node.Val)
} }

View File

@ -0,0 +1,98 @@
// File: print_utils.go
// Created Time: 2022-12-03
// Author: Reanon (793584285@qq.com), Krahets (krahets@163.com)
package pkg
import (
"container/list"
"fmt"
"strconv"
"strings"
)
func PrintSlice(nums []int) {
fmt.Printf("%v", nums)
fmt.Println()
}
// PrintList Print a list
func PrintList(list *list.List) {
e := list.Front()
// 强转为 string, 会影响效率
fmt.Print("[")
for e.Next() != nil {
fmt.Print(e.Value, " ")
e = e.Next()
}
fmt.Print(e.Value, "]\n")
}
// PrintLinkedList Print a linked list
func PrintLinkedList(node *ListNode) {
if node == nil {
return
}
var builder strings.Builder
for node.Next != nil {
builder.WriteString(strconv.Itoa(node.Val) + " -> ")
node = node.Next
}
builder.WriteString(strconv.Itoa(node.Val))
fmt.Println(builder.String())
}
// PrintTree Print a binary tree
func PrintTree(root *TreeNode) {
printTreeHelper(root, nil, false)
}
// printTreeHelper Help to print a binary tree, hide more details
// This tree printer is borrowed from TECHIE DELIGHT
// https://www.techiedelight.com/c-program-print-binary-tree/
func printTreeHelper(root *TreeNode, prev *trunk, isLeft bool) {
if root == nil {
return
}
prevStr := " "
trunk := newTrunk(prev, prevStr)
printTreeHelper(root.Right, trunk, true)
if prev == nil {
trunk.str = "———"
} else if isLeft {
trunk.str = "/———"
prevStr = " |"
} else {
trunk.str = "\\———"
prev.str = prevStr
}
showTrunk(trunk)
fmt.Println(root.Val)
if prev != nil {
prev.str = prevStr
}
trunk.str = " |"
printTreeHelper(root.Left, trunk, false)
}
// trunk Help to Print tree structure
type trunk struct {
prev *trunk
str string
}
func newTrunk(prev *trunk, str string) *trunk {
return &trunk{
prev: prev,
str: str,
}
}
func showTrunk(t *trunk) {
if t == nil {
return
}
showTrunk(t.prev)
fmt.Print(t.str)
}

View File

@ -6,7 +6,6 @@ package pkg
import ( import (
"container/list" "container/list"
"fmt"
) )
type TreeNode struct { type TreeNode struct {
@ -71,58 +70,3 @@ func TreeToArray(root *TreeNode) []any {
} }
return arr return arr
} }
// PrintTree Print a binary tree
func PrintTree(root *TreeNode) {
printTreeHelper(root, nil, false)
}
// printTreeHelper Help to print a binary tree, hide more details
// This tree printer is borrowed from TECHIE DELIGHT
// https://www.techiedelight.com/c-program-print-binary-tree/
func printTreeHelper(root *TreeNode, prev *trunk, isLeft bool) {
if root == nil {
return
}
prevStr := " "
trunk := newTrunk(prev, prevStr)
printTreeHelper(root.Right, trunk, true)
if prev == nil {
trunk.str = "———"
} else if isLeft {
trunk.str = "/———"
prevStr = " |"
} else {
trunk.str = "\\———"
prev.str = prevStr
}
showTrunk(trunk)
fmt.Println(root.Val)
if prev != nil {
prev.str = prevStr
}
trunk.str = " |"
printTreeHelper(root.Left, trunk, false)
}
// trunk Help to Print tree structure
type trunk struct {
prev *trunk
str string
}
func newTrunk(prev *trunk, str string) *trunk {
return &trunk{
prev: prev,
str: str,
}
}
func showTrunk(t *trunk) {
if t == nil {
return
}
showTrunk(t.prev)
fmt.Print(t.str)
}

View File

@ -4,7 +4,10 @@
package pkg package pkg
import "testing" import (
"fmt"
"testing"
)
func TestTreeNode(t *testing.T) { func TestTreeNode(t *testing.T) {
arr := []int{2, 3, 5, 6, 7} arr := []int{2, 3, 5, 6, 7}
@ -14,5 +17,5 @@ func TestTreeNode(t *testing.T) {
PrintTree(node) PrintTree(node)
// tree to arr // tree to arr
t.Log(TreeToArray(node)) fmt.Println(TreeToArray(node))
} }

View File

@ -21,7 +21,7 @@ public class deque {
deque.offerFirst(1); deque.offerFirst(1);
System.out.println("双向队列 deque = " + deque); System.out.println("双向队列 deque = " + deque);
/* 访问队首元素 */ /* 访问元素 */
int peekFirst = deque.peekFirst(); int peekFirst = deque.peekFirst();
System.out.println("队首元素 peekFirst = " + peekFirst); System.out.println("队首元素 peekFirst = " + peekFirst);
int peekLast = deque.peekLast(); int peekLast = deque.peekLast();

View File

@ -7,7 +7,7 @@
/* 随机访问元素 */ /* 随机访问元素 */
function randomAccess(nums){ function randomAccess(nums){
// 在区间 [0, nums.length) 中随机抽取一个数字 // 在区间 [0, nums.length) 中随机抽取一个数字
const random_index = Math.floor(Math.random() * nums.length) const random_index = Math.floor(Math.random() * nums.length)
// 获取并返回随机元素 // 获取并返回随机元素
random_num = nums[random_index] random_num = nums[random_index]
return random_num return random_num

View File

@ -21,4 +21,4 @@ function insertionSort(nums) {
/* Driver Code */ /* Driver Code */
var nums = [4, 1, 3, 1, 5, 2] var nums = [4, 1, 3, 1, 5, 2]
insertionSort(nums) insertionSort(nums)
console.log("排序后数组 nums = ", nums) console.log("排序后数组 nums =", nums)

View File

@ -33,7 +33,7 @@ if __name__ == "__main__":
list.append(2) list.append(2)
list.append(5) list.append(5)
list.append(4) list.append(4)
print("添加元素后 list = ", list) print("添加元素后 list =", list)
""" 中间插入元素 """ """ 中间插入元素 """
list.insert(3, 6) list.insert(3, 6)

View File

@ -1,10 +1,53 @@
''' '''
File: binary_search.py File: binary_search.py
Created Time: 2022-11-25 Created Time: 2022-11-26
Author: Krahets (krahets@163.com) Author: timi (xisunyy@163.com)
''' '''
import sys, os.path as osp import sys, os.path as osp
sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__)))) sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__))))
from include import * from include import *
""" 二分查找(双闭区间) """
def binary_search(nums, target):
# 初始化双闭区间 [0, n-1] ,即 i, j 分别指向数组首元素、尾元素
i, j = 0, len(nums) - 1
while i <= j:
m = (i + j) // 2 # 计算中点索引 m
if nums[m] < target: # 此情况说明 target 在区间 [m+1, j] 中
i = m + 1
elif nums[m] > target: # 此情况说明 target 在区间 [i, m-1] 中
j = m - 1
else:
return m # 找到目标元素,返回其索引
return -1 # 未找到目标元素,返回 -1
""" 二分查找(左闭右开) """
def binary_search1(nums, target):
# 初始化左闭右开 [0, n) ,即 i, j 分别指向数组首元素、尾元素+1
i, j = 0, len(nums)
# 循环,当搜索区间为空时跳出(当 i = j 时为空)
while i < j:
m = (i + j) // 2 # 计算中点索引 m
if nums[m] < target: # 此情况说明 target 在区间 [m+1, j) 中
i = m + 1
elif nums[m] > target: # 此情况说明 target 在区间 [i, m) 中
j = m
else: # 找到目标元素,返回其索引
return m
return -1 # 未找到目标元素,返回 -1
""" Driver Code """
if __name__ == '__main__':
target = 6
nums = [1, 3, 6, 8, 12, 15, 23, 67, 70, 92]
# 二分查找(双闭区间)
index = binary_search(nums, target)
print("目标元素 6 的索引 = ", index)
# 二分查找(左闭右开)
index = binary_search1(nums, target)
print("目标元素 6 的索引 = ", index)

View File

@ -1,10 +1,45 @@
''' '''
File: hashing_search.py File: hashing_search.py
Created Time: 2022-11-25 Created Time: 2022-11-26
Author: Krahets (krahets@163.com) Author: timi (xisunyy@163.com)
''' '''
import sys, os.path as osp import sys, os.path as osp
sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__)))) sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__))))
from include import * from include import *
""" 哈希查找(数组) """
def hashing_search(mapp, target):
# 哈希表的 key: 目标元素value: 索引
# 若哈希表中无此 key ,返回 -1
return mapp.get(target, -1)
""" 哈希查找(链表) """
def hashing_search1(mapp, target):
# 哈希表的 key: 目标元素value: 结点对象
# 若哈希表中无此 key ,返回 -1
return mapp.get(target, -1)
""" Driver Code """
if __name__ == '__main__':
target = 3
# 哈希查找(数组)
nums = [1, 5, 3, 2, 4, 7, 5, 9, 10, 8]
# 初始化哈希表
mapp = {}
for i in range(len(nums)):
mapp[nums[i]] = i # key: 元素value: 索引
index = hashing_search(mapp, target)
print("目标元素 3 的索引 =", index)
# 哈希查找(链表)
head = list_to_linked_list(nums)
# 初始化哈希表
map1 = {}
while head:
map1[head.val] = head # key: 结点值value: 结点
head = head.next
node = hashing_search1(map1, target)
print("目标结点值 3 的对应结点对象为", node)

View File

@ -1,10 +1,41 @@
''' '''
File: linear_search.py File: linear_search.py
Created Time: 2022-11-25 Created Time: 2022-11-26
Author: Krahets (krahets@163.com) Author: timi (xisunyy@163.com)
''' '''
import sys, os.path as osp import sys, os.path as osp
sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__)))) sys.path.append(osp.dirname(osp.dirname(osp.abspath(__file__))))
from include import * from include import *
""" 线性查找(数组) """
def linear_search(nums, target):
# 遍历数组
for i in range(len(nums)):
if nums[i] == target: # 找到目标元素,返回其索引
return i
return -1 # 未找到目标元素,返回 -1
""" 线性查找(链表) """
def linear_search1(head, target):
# 遍历链表
while head:
if head.val == target: # 找到目标结点,返回之
return head
head = head.next
return None # 未找到目标结点,返回 None
""" Driver Code """
if __name__ == '__main__':
target = 3
# 在数组中执行线性查找
nums = [1, 5, 3, 2, 4, 7, 5, 9, 10, 8]
index = linear_search(nums, target)
print("目标元素 3 的索引 =", index)
# 在链表中执行线性查找
head = list_to_linked_list(nums)
node = linear_search1(head, target)
print("目标结点值 3 的对应结点对象为", node)

View File

@ -25,4 +25,4 @@ def insertion_sort(nums):
if __name__ == '__main__': if __name__ == '__main__':
nums = [4, 1, 3, 1, 5, 2] nums = [4, 1, 3, 1, 5, 2]
insertion_sort(nums) insertion_sort(nums)
print("排序后数组 nums = ", nums) print("排序后数组 nums =", nums)

View File

@ -26,7 +26,7 @@ if __name__ == "__main__":
duque.appendleft(1) duque.appendleft(1)
print("双向队列 duque =", duque) print("双向队列 duque =", duque)
""" 访问队首元素 """ """ 访问元素 """
front = duque[0] # 队首元素 front = duque[0] # 队首元素
print("队首元素 front =", front) print("队首元素 front =", front)
rear = duque[-1] # 队尾元素 rear = duque[-1] # 队尾元素

View File

@ -138,7 +138,7 @@ elementAddr = firtstElementAddr + elementLength * elementIndex
/* 随机返回一个数组元素 */ /* 随机返回一个数组元素 */
function randomAccess(nums){ function randomAccess(nums){
// 在区间 [0, nums.length) 中随机抽取一个数字 // 在区间 [0, nums.length) 中随机抽取一个数字
const random_index = Math.floor(Math.random() * nums.length) const random_index = Math.floor(Math.random() * nums.length)
// 获取并返回随机元素 // 获取并返回随机元素
random_num = nums[random_index] random_num = nums[random_index]
return random_num return random_num

View File

@ -520,7 +520,7 @@ $$
```python title="space_complexity_types.py" ```python title="space_complexity_types.py"
""" 线性阶(递归实现) """ """ 线性阶(递归实现) """
def linearRecur(n): def linearRecur(n):
print("递归 n = ", n) print("递归 n =", n)
if n == 1: return if n == 1: return
linearRecur(n - 1) linearRecur(n - 1)
``` ```

View File

@ -15,7 +15,7 @@ comments: true
例如以下代码,输入数据大小为 $n$ ,根据以上方法,可以得到算法运行时间为 $6n + 12$ ns 。 例如以下代码,输入数据大小为 $n$ ,根据以上方法,可以得到算法运行时间为 $6n + 12$ ns 。
$$ $$
1 + 1 + 10 + (1 + 5) \times n = 6n + 12 1 + 1 + 10 + (1 + 5) \times n = 6n + 12
$$ $$
=== "Java" === "Java"

View File

@ -105,7 +105,19 @@ $$
=== "Python" === "Python"
```python title="binary_search.py" ```python title="binary_search.py"
""" 二分查找(双闭区间) """
def binary_search(nums, target):
# 初始化双闭区间 [0, n-1] ,即 i, j 分别指向数组首元素、尾元素
i, j = 0, len(nums) - 1
while i <= j:
m = (i + j) // 2 # 计算中点索引 m
if nums[m] < target: # 此情况说明 target 在区间 [m+1, j]
i = m + 1
elif nums[m] > target: # 此情况说明 target 在区间 [i, m-1] 中
j = m - 1
else:
return m # 找到目标元素,返回其索引
return -1 # 未找到目标元素,返回 -1
``` ```
=== "Go" === "Go"
@ -189,7 +201,20 @@ $$
=== "Python" === "Python"
```python title="binary_search.py" ```python title="binary_search.py"
""" 二分查找(左闭右开) """
def binary_search1(nums, target):
# 初始化左闭右开 [0, n) ,即 i, j 分别指向数组首元素、尾元素+1
i, j = 0, len(nums)
# 循环,当搜索区间为空时跳出(当 i = j 时为空)
while i < j:
m = (i + j) // 2 # 计算中点索引 m
if nums[m] < target: # 此情况说明 target 在区间 [m+1, j)
i = m + 1
elif nums[m] > target: # 此情况说明 target 在区间 [i, m) 中
j = m
else: # 找到目标元素,返回其索引
return m
return -1 # 未找到目标元素,返回 -1
``` ```
=== "Go" === "Go"
@ -262,8 +287,8 @@ $$
=== "Python" === "Python"
```py title="" ```py title=""
# Python 中的数字理论上可以无限大(取决于内存) # Python 中的数字理论上可以无限大(取决于内存大小
# 因此无需考虑大数越界问题 # 因此无需考虑大数越界问题
``` ```
=== "Go" === "Go"

View File

@ -43,7 +43,11 @@ comments: true
=== "Python" === "Python"
```python title="hashing_search.py" ```python title="hashing_search.py"
""" 哈希查找(数组) """
def hashing_search(mapp, target):
# 哈希表的 key: 目标元素value: 索引
# 若哈希表中无此 key ,返回 -1
return mapp.get(target, -1)
``` ```
=== "Go" === "Go"
@ -107,7 +111,11 @@ comments: true
=== "Python" === "Python"
```python title="hashing_search.py" ```python title="hashing_search.py"
""" 哈希查找(链表) """
def hashing_search1(mapp, target):
# 哈希表的 key: 目标元素value: 结点对象
# 若哈希表中无此 key ,返回 -1
return mapp.get(target, -1)
``` ```
=== "Go" === "Go"

View File

@ -47,7 +47,13 @@ comments: true
=== "Python" === "Python"
```python title="linear_search.py" ```python title="linear_search.py"
""" 线性查找(数组) """
def linear_search(nums, target):
# 遍历数组
for i in range(len(nums)):
if nums[i] == target: # 找到目标元素,返回其索引
return i
return -1 # 未找到目标元素,返回 -1
``` ```
=== "Go" === "Go"
@ -119,7 +125,14 @@ comments: true
=== "Python" === "Python"
```python title="linear_search.py" ```python title="linear_search.py"
""" 线性查找(链表) """
def linear_search1(head, target):
# 遍历链表
while head:
if head.val == target: # 找到目标结点,返回之
return head
head = head.next
return None # 未找到目标结点,返回 None
``` ```
=== "Go" === "Go"

View File

@ -46,7 +46,7 @@ comments: true
deque.offerFirst(3); // 添加至队首 deque.offerFirst(3); // 添加至队首
deque.offerFirst(1); deque.offerFirst(1);
/* 访问队首元素 */ /* 访问元素 */
int peekFirst = deque.peekFirst(); // 队首元素 int peekFirst = deque.peekFirst(); // 队首元素
int peekLast = deque.peekLast(); // 队尾元素 int peekLast = deque.peekLast(); // 队尾元素
@ -74,7 +74,7 @@ comments: true
deque.push_front(3); // 添加至队首 deque.push_front(3); // 添加至队首
deque.push_front(1); deque.push_front(1);
/* 访问队首元素 */ /* 访问元素 */
int front = deque.front(); // 队首元素 int front = deque.front(); // 队首元素
int back = deque.back(); // 队尾元素 int back = deque.back(); // 队尾元素
@ -102,7 +102,7 @@ comments: true
duque.appendleft(3) # 添加至队首 duque.appendleft(3) # 添加至队首
duque.appendleft(1) duque.appendleft(1)
""" 访问队首元素 """ """ 访问元素 """
front = duque[0] # 队首元素 front = duque[0] # 队首元素
rear = duque[-1] # 队尾元素 rear = duque[-1] # 队尾元素
@ -120,7 +120,30 @@ comments: true
=== "Go" === "Go"
```go title="deque.go" ```go title="deque.go"
/* 初始化双向队列 */
// 在 Go 中,将 list 作为双向队列使用
deque := list.New()
/* 元素入队 */
deque.PushBack(2) // 添加至队尾
deque.PushBack(5)
deque.PushBack(4)
deque.PushFront(3) // 添加至队首
deque.PushFront(1)
/* 访问元素 */
front := deque.Front() // 队首元素
rear := deque.Back() // 队尾元素
/* 元素出队 */
deque.Remove(front) // 队首元素出队
deque.Remove(rear) // 队尾元素出队
/* 获取双向队列的长度 */
size := deque.Len()
/* 判断双向队列是否为空 */
isEmpty := deque.Len() == 0
``` ```
=== "JavaScript" === "JavaScript"

View File

@ -115,7 +115,29 @@ comments: true
=== "Go" === "Go"
```go title="queue.go" ```go title="queue.go"
/* 初始化队列 */
// 在 Go 中,将 list 作为队列来使用
queue := list.New()
/* 元素入队 */
queue.PushBack(1)
queue.PushBack(3)
queue.PushBack(2)
queue.PushBack(5)
queue.PushBack(4)
/* 访问队首元素 */
peek := queue.Front()
/* 元素出队 */
poll := queue.Front()
queue.Remove(poll)
/* 获取队列的长度 */
size := queue.Len()
/* 判断队列是否为空 */
isEmpty := queue.Len() == 0
``` ```
=== "JavaScript" === "JavaScript"
@ -309,7 +331,52 @@ comments: true
=== "Go" === "Go"
```go title="linkedlist_queue.go" ```go title="linkedlist_queue.go"
/* 基于链表实现的队列 */
type LinkedListQueue struct {
// 使用内置包 list 来实现队列
data *list.List
}
// NewLinkedListQueue 初始化链表
func NewLinkedListQueue() *LinkedListQueue {
return &LinkedListQueue{
data: list.New(),
}
}
// Offer 入队
func (s *LinkedListQueue) Offer(value any) {
s.data.PushBack(value)
}
// Poll 出队
func (s *LinkedListQueue) Poll() any {
if s.IsEmpty() {
return nil
}
e := s.data.Front()
s.data.Remove(e)
return e.Value
}
// Peek 访问队首元素
func (s *LinkedListQueue) Peek() any {
if s.IsEmpty() {
return nil
}
e := s.data.Front()
return e.Value
}
// Size 获取队列的长度
func (s *LinkedListQueue) Size() int {
return s.data.Len()
}
// IsEmpty 判断队列是否为空
func (s *LinkedListQueue) IsEmpty() bool {
return s.data.Len() == 0
}
``` ```
=== "JavaScript" === "JavaScript"
@ -540,7 +607,66 @@ comments: true
=== "Go" === "Go"
```go title="array_queue.go" ```go title="array_queue.go"
/* 基于环形数组实现的队列 */
type ArrayQueue struct {
data []int // 用于存储队列元素的数组
capacity int // 队列容量(即最多容量的元素个数)
front int // 头指针,指向队首
rear int // 尾指针,指向队尾 + 1
}
// NewArrayQueue 基于环形数组实现的队列
func NewArrayQueue(capacity int) *ArrayQueue {
return &ArrayQueue{
data: make([]int, capacity),
capacity: capacity,
front: 0,
rear: 0,
}
}
// Size 获取队列的长度
func (q *ArrayQueue) Size() int {
size := (q.capacity + q.rear - q.front) % q.capacity
return size
}
// IsEmpty 判断队列是否为空
func (q *ArrayQueue) IsEmpty() bool {
return q.rear-q.front == 0
}
// Offer 入队
func (q *ArrayQueue) Offer(v int) {
// 当 rear == capacity 表示队列已满
if q.Size() == q.capacity {
return
}
// 尾结点后添加
q.data[q.rear] = v
// 尾指针向后移动一位,越过尾部后返回到数组头部
q.rear = (q.rear + 1) % q.capacity
}
// Poll 出队
func (q *ArrayQueue) Poll() any {
if q.IsEmpty() {
return nil
}
v := q.data[q.front]
// 队头指针向后移动一位,若越过尾部则返回到数组头部
q.front = (q.front + 1) % q.capacity
return v
}
// Peek 访问队首元素
func (q *ArrayQueue) Peek() any {
if q.IsEmpty() {
return nil
}
v := q.data[q.front]
return v
}
``` ```
=== "JavaScript" === "JavaScript"

View File

@ -115,7 +115,29 @@ comments: true
=== "Go" === "Go"
```go title="stack.go" ```go title="stack.go"
/* 初始化栈 */
// 在 Go 中,推荐将 Slice 当作栈来使用
var stack []int
/* 元素入栈 */
stack = append(stack, 1)
stack = append(stack, 3)
stack = append(stack, 2)
stack = append(stack, 5)
stack = append(stack, 4)
/* 访问栈顶元素 */
peek := stack[len(stack)-1]
/* 元素出栈 */
pop := stack[len(stack)-1]
stack = stack[:len(stack)-1]
/* 获取栈的长度 */
size := len(stack)
/* 判断是否为空 */
isEmpty := len(stack) == 0
``` ```
=== "JavaScript" === "JavaScript"
@ -282,7 +304,52 @@ comments: true
=== "Go" === "Go"
```go title="linkedlist_stack.go" ```go title="linkedlist_stack.go"
/* 基于链表实现的栈 */
type LinkedListStack struct {
// 使用内置包 list 来实现栈
data *list.List
}
// NewLinkedListStack 初始化链表
func NewLinkedListStack() *LinkedListStack {
return &LinkedListStack{
data: list.New(),
}
}
// Push 入栈
func (s *LinkedListStack) Push(value int) {
s.data.PushBack(value)
}
// Pop 出栈
func (s *LinkedListStack) Pop() any {
if s.IsEmpty() {
return nil
}
e := s.data.Back()
s.data.Remove(e)
return e.Value
}
// Peek 访问栈顶元素
func (s *LinkedListStack) Peek() any {
if s.IsEmpty() {
return nil
}
e := s.data.Back()
return e.Value
}
// Size 获取栈的长度
func (s *LinkedListStack) Size() int {
return s.data.Len()
}
// IsEmpty 判断栈是否为空
func (s *LinkedListStack) IsEmpty() bool {
return s.data.Len() == 0
}
``` ```
=== "JavaScript" === "JavaScript"
@ -426,7 +493,53 @@ comments: true
=== "Go" === "Go"
```go title="array_stack.go" ```go title="array_stack.go"
/* 基于数组实现的栈 */
type ArrayStack struct {
data []int // 数据
}
func NewArrayStack() *ArrayStack {
return &ArrayStack{
// 设置栈的长度为 0容量为 16
data: make([]int, 0, 16),
}
}
// Size 栈的长度
func (s *ArrayStack) Size() int {
return len(s.data)
}
// IsEmpty 栈是否为空
func (s *ArrayStack) IsEmpty() bool {
return s.Size() == 0
}
// Push 入栈
func (s *ArrayStack) Push(v int) {
// 切片会自动扩容
s.data = append(s.data, v)
}
// Pop 出栈
func (s *ArrayStack) Pop() any {
// 弹出栈前,先判断是否为空
if s.IsEmpty() {
return nil
}
val := s.Peek()
s.data = s.data[:len(s.data)-1]
return val
}
// Peek 获取栈顶元素
func (s *ArrayStack) Peek() any {
if s.IsEmpty() {
return nil
}
val := s.data[len(s.data)-1]
return val
}
``` ```
=== "JavaScript" === "JavaScript"

View File

@ -88,7 +88,25 @@ comments: true
=== "Go" === "Go"
```go title="binary_search_tree.go" ```go title="binary_search_tree.go"
/* 查找结点 */
func (bst *BinarySearchTree) Search(num int) *TreeNode {
node := bst.root
// 循环查找,越过叶结点后跳出
for node != nil {
if node.Val < num {
// 目标结点在 root 的右子树中
node = node.Right
} else if node.Val > num {
// 目标结点在 root 的左子树中
node = node.Left
} else {
// 找到目标结点,跳出循环
break
}
}
// 返回目标结点
return node
}
``` ```
=== "JavaScript" === "JavaScript"
@ -187,7 +205,36 @@ comments: true
=== "Go" === "Go"
```go title="binary_search_tree.go" ```go title="binary_search_tree.go"
/* 插入结点 */
func (bst *BinarySearchTree) Insert(num int) *TreeNode {
cur := bst.root
// 若树为空,直接提前返回
if cur == nil {
return nil
}
// 待插入结点之前的结点位置
var prev *TreeNode = nil
// 循环查找,越过叶结点后跳出
for cur != nil {
if cur.Val == num {
return nil
}
prev = cur
if cur.Val < num {
cur = cur.Right
} else {
cur = cur.Left
}
}
// 插入结点
node := NewTreeNode(num)
if prev.Val < num {
prev.Right = node
} else {
prev.Left = node
}
return cur
}
``` ```
=== "JavaScript" === "JavaScript"
@ -365,7 +412,60 @@ comments: true
=== "Go" === "Go"
```go title="binary_search_tree.go" ```go title="binary_search_tree.go"
/* 删除结点 */
func (bst *BinarySearchTree) Remove(num int) *TreeNode {
cur := bst.root
// 若树为空,直接提前返回
if cur == nil {
return nil
}
// 待删除结点之前的结点位置
var prev *TreeNode = nil
// 循环查找,越过叶结点后跳出
for cur != nil {
if cur.Val == num {
break
}
prev = cur
if cur.Val < num {
// 待删除结点在右子树中
cur = cur.Right
} else {
// 待删除结点在左子树中
cur = cur.Left
}
}
// 若无待删除结点,则直接返回
if cur == nil {
return nil
}
// 子结点数为 0 或 1
if cur.Left == nil || cur.Right == nil {
var child *TreeNode = nil
// 取出待删除结点的子结点
if cur.Left != nil {
child = cur.Left
} else {
child = cur.Right
}
// 将子结点替换为待删除结点
if prev.Left == cur {
prev.Left = child
} else {
prev.Right = child
}
// 子结点数为 2
} else {
// 获取中序遍历中待删除结点 cur 的下一个结点
next := bst.GetInorderNext(cur)
temp := next.Val
// 递归删除结点 next
bst.Remove(next.Val)
// 将 next 的值复制给 cur
cur.Val = temp
}
return cur
}
``` ```
=== "JavaScript" === "JavaScript"

View File

@ -44,7 +44,20 @@ comments: true
=== "Go" === "Go"
```go title="" ```go title=""
""" 链表结点类 """
type TreeNode struct {
Val int
Left *TreeNode
Right *TreeNode
}
""" 结点初始化方法 """
func NewTreeNode(v int) *TreeNode {
return &TreeNode{
Left: nil,
Right: nil,
Val: v,
}
}
``` ```
=== "JavaScript" === "JavaScript"
@ -163,7 +176,18 @@ comments: true
=== "Go" === "Go"
```go title="binary_tree.go" ```go title="binary_tree.go"
/* 初始化二叉树 */
// 初始化结点
n1 := NewTreeNode(1)
n2 := NewTreeNode(2)
n3 := NewTreeNode(3)
n4 := NewTreeNode(4)
n5 := NewTreeNode(5)
// 构建引用指向(即指针)
n1.Left = n2
n1.Right = n3
n2.Left = n4
n2.Right = n5
``` ```
=== "JavaScript" === "JavaScript"
@ -228,7 +252,14 @@ comments: true
=== "Go" === "Go"
```go title="binary_tree.go" ```go title="binary_tree.go"
/* 插入与删除结点 */
// 在 n1 -> n2 中间插入结点 P
p := NewTreeNode(0)
n1.Left = p
p.Left = n2
// 删除结点 P
n1.Left = n2
``` ```
=== "JavaScript" === "JavaScript"
@ -328,7 +359,29 @@ comments: true
=== "Go" === "Go"
```go title="binary_tree_bfs.go" ```go title="binary_tree_bfs.go"
/* 层序遍历 */
func levelOrder(root *TreeNode) []int {
// 初始化队列,加入根结点
queue := list.New()
queue.PushBack(root)
// 初始化一个切片,用于保存遍历序列
nums := make([]int, 0)
for queue.Len() > 0 {
// poll
node := queue.Remove(queue.Front()).(*TreeNode)
// 保存结点
nums = append(nums, node.Val)
if node.Left != nil {
// 左子结点入队
queue.PushBack(node.Left)
}
if node.Right != nil {
// 右子结点入队
queue.PushBack(node.Right)
}
}
return nums
}
``` ```
=== "JavaScript" === "JavaScript"
@ -446,7 +499,38 @@ comments: true
=== "Go" === "Go"
```go title="binary_tree_dfs.go" ```go title="binary_tree_dfs.go"
/* 前序遍历 */
func preOrder(node *TreeNode) {
if node == nil {
return
}
// 访问优先级:根结点 -> 左子树 -> 右子树
nums = append(nums, node.Val)
preOrder(node.Left)
preOrder(node.Right)
}
/* 中序遍历 */
func inOrder(node *TreeNode) {
if node == nil {
return
}
// 访问优先级:左子树 -> 根结点 -> 右子树
inOrder(node.Left)
nums = append(nums, node.Val)
inOrder(node.Right)
}
/* 后序遍历 */
func postOrder(node *TreeNode) {
if node == nil {
return
}
// 访问优先级:左子树 -> 右子树 -> 根结点
postOrder(node.Left)
postOrder(node.Right)
nums = append(nums, node.Val)
}
``` ```
=== "JavaScript" === "JavaScript"