golang深度获取子节点

起因

需要在树形结构里获取子树，树形结构一般是存储一维数组，数组元素里存储子节点的指针

代码

package main

import (
	"errors"
	"fmt"
)

type Node struct {
	Deep  int
	Child *Node
}

func (n *Node) GetChild() *Node {
	return n.Child
}

func (n *Node) SetChild(child *Node) {
	n.Child = child
}

func GetDeepChild(node *Node, level int) (*Node, error) {
	if level < 0 {
		return nil, errors.New("level must >= 0")
	}
	if level == 0 {
		return node, nil
	} else {
		child := node.GetChild()
		if child == nil {
			return nil, nil
		}
		return GetDeepChild(child, level-1)
	}

}

func main() {
	root := &Node{Deep: 0}
	child1 := &Node{Deep: 1}
	root.SetChild(child1)

	child2 := &Node{Deep: 2}
	child1.SetChild(child2)

	child3 := &Node{Deep: 3}
	child2.SetChild(child3)

	child4 := &Node{Deep: 4}
	child3.SetChild(child4)

	child, _ := GetDeepChild(root, 3)
	fmt.Printf("child %#v
", child) // deep 3
	child, _ = GetDeepChild(root, 5)
	fmt.Printf("child %#v
", child) // nil
}