【Golang 面试算法二叉树 手动建树】

前言

面试中有很多有关二叉树的题目，且需要手动建树，这里记录一下如何用Golang来快速构建一个二叉树。

代码实现

我们知道二叉树可以扁平化到数组中，当前节点的左右子节点的在数组中的下标可以表示为

leftIndex := currIndex * 2 + 1

rightIndex := currIndex * 2 + 2

其中currIndex表示当前节点在数组中的下标，基于此我们可以构建二叉树。

Node节点数据结构

type Node struct {
   
   
	Val         int
	Left, Right *Node
}

buildTree方法

// buildTree 构建树并且返回树的头节点
// arr 树节点的扁平化数组，其中-1表示空节点
// index 当前节点在数组中的下标
// n 数组长度
func buildTree(arr []int, index, n int) *Node {
   
   
	if n == 0 || index >= n || arr[index