golang编写二叉树
最近开始找golang 开发工程师职位，针对算法相关二叉树相关常用面试题搞一遍：
package tree
import (
"math"
"fmt"
)
type BinaryTree struct {
Value int
Left *BinaryTree
Right *BinaryTree
}
func InitBinaryTree(root *BinaryTree) *BinaryTree {
l := BinaryTree{}
r := BinaryTree{}
root.Left = l.NewBinaryTreeNode(2)
root.Right = r.NewBinaryTreeNode(3)
l2:=BinaryTree{}
ll2:=BinaryTree{}
root.Left.Left = l2.NewBinaryTreeNode(4)
root.Left.Right = ll2.NewBinaryTreeNode(5)
return root
}
func (this *BinaryTree) NewBinaryTreeNode(value int) *BinaryTree {
this.Value = value
this.Left = nil
this.Right = nil
return this
}
// 计算二叉树节点个数
func GetNodeNum(root *BinaryTree) int {
if root == nil {
return 0
} else {
return GetNodeNum(root.Left) + GetNodeNum(root.Right) + 1
}
}
// 计算二叉树的深度
func GetDegree(root *BinaryTree) int {
if root == nil {
return 0
}
var maxDegree = 0
if GetDegree(root.Left) > GetDegree(root.Right) {
maxDegree = GetDegree(root.Left)
} else {
maxDegree = GetDegree(root.Right)
}
return maxDegree + 1
}
// 前序遍历： 根-> 左子树 -> 右子树
func PreOrder(root *BinaryTree) {
if root == nil {
return
}
fmt.Printf("%d->", root.Value)
PreOrder(root.Left)
PreOrder(root.Right)
}
// 中序： 左子树-> 根 -> 右子树
func InOrder(root *BinaryTree) {
if root == nil {
return
}
InOrder(root.Left)
fmt.Printf("%d->", root.Value)
InOrder(root.Right)
}
// 后序： 左子树-> 右子树 -> 根
func PostOrder(root *BinaryTree) {
if root == nil {
return
}
PostOrder(root.Left)
PostOrder(root.Right)
fmt.Printf("%d->", root.Value)
}
// 求 K 层节点个数
func GetKthNum(root *BinaryTree ,k int ) int {
if root==nil{
return 0
}
if k==1{
return 1
}
return GetKthNum(root.Left,k-1)+GetKthNum(root.Right,k-1)
}
// 求叶子节点个数
func GetLeavNum(root *BinaryTree) int {
if root==nil{
return 0
}
if root.Left==nil && root.Right == nil {
return 1
}
return GetLeavNum(root.Left)+GetLeavNum(root.Right)
}
// 判断是否平衡二叉树
func IsBalanced(root *BinaryTree) bool {
if root==nil{
return true
}
lde:=GetDegree(root.Left)
rde:=GetDegree(root.Right)
flag:=false
if (math.Abs(float64(lde-rde)))<=1{
flag=true
}else{
flag=false
}
return flag && IsBalanced(root.Left) && IsBalanced(root.Right)
}
