Golang的树的实现

golang的树结构的简单实现，未完善，代码如下：

package main

import (
    "fmt"
    "errors"
)

func main() {
    var t Tree
    t.Create("美利坚合众国")
    t.AddSon("国土安全部")
    t.AddSon("财政部")
    t.AddSon("信息科技部")
    t.AddSon("能源部")
    t.AddSon("军事委员会")
    fmt.Println(t.GetName(), "共有", t.GetDegree(), "个部门")
    fmt.Println("------------------------------------------")
    for k, v := range t.son {
        fmt.Println(t.GetName(), "第", k, "个部门是", v.GetName())
    }

    s,err := t.GetSon("国土安全部")
    if err == nil {
        s.AddSon("质量安全办")
        s.AddSon("征地拆迁办")
    }
    fmt.Println("==========================================")
    fmt.Println(s.GetName(), "共有", s.GetDegree(), "个部门")
    fmt.Println("------------------------------------------")
    for k, v := range s.son {
        fmt.Println(s.GetName(), "第", k, "个部门是", v.GetName())
    }

    s1,err := s.GetSon("质量安全办")
    if err == nil {
        s1.AddSon("特朗普")
        s1.AddSon("蓬佩奥")
        s1.AddSon("拜登")
        s1.AddSon("邓文迪")
        s1.AddSon("李冰冰")
    }
    fmt.Println("==========================================")
    fmt.Println(s1.GetName(), "共有", s1.GetDegree(), "名员工")
    fmt.Println("------------------------------------------")
    for k, v := range s1.son {
        fmt.Println(s1.GetName(), "第", k, "名员工是", v.GetName())
    }
    s1.DeleteSon("蓬佩奥")
    fmt.Println("==========================================")
    fmt.Println(s1.GetName(), "共有", s1.GetDegree(), "名员工")
    fmt.Println("------------------------------------------")
    for k, v := range s1.son {
        fmt.Println(s1.GetName(), "第", k, "名员工是", v.GetName())
    }
}

type Tree struct {
    name   string
    parent *Tree
    son    []*Tree
    data   interface{}
}

func (this *Tree) Create(name string) {
    this.name = name
    this.son = nil
}

func (this *Tree) GetName() string {
    return this.name
}

func (this *Tree) GetSon(nameofson string) (son *Tree, err error) {
    for _, v := range this.son {
        if v.GetName() == nameofson {
            return v,nil
        }
    }
    return nil, errors.New("No son's name call" + nameofson)
}

func (this *Tree) AddSon(nameofson string) {
    var Son Tree

    Son.name = nameofson
    Son.parent = this
    this.son = append(this.son, &Son)
}

func (this *Tree) DeleteSon(nameofson string) {
    for k, v := range this.son {
        if v.GetName() == nameofson {
            this.son = append(this.son[:k], this.son[(k+1):]...)
        }
    }
}

func (this Tree) GetDegree() int {
    return len(this.son)
}

运行结果如下：

美利坚合众国 共有 5 个部门
------------------------------------------
美利坚合众国 第 0 个部门是 国土安全部
美利坚合众国 第 1 个部门是 财政部
美利坚合众国 第 2 个部门是 信息科技部
美利坚合众国 第 3 个部门是 能源部
美利坚合众国 第 4 个部门是 军事委员会
==========================================
国土安全部 共有 2 个部门
------------------------------------------
国土安全部 第 0 个部门是 质量安全办
国土安全部 第 1 个部门是 征地拆迁办
==========================================
质量安全办 共有 5 名员工
------------------------------------------
质量安全办 第 0 名员工是 特朗普
质量安全办 第 1 名员工是 蓬佩奥
质量安全办 第 2 名员工是 拜登
质量安全办 第 3 名员工是 邓文迪
质量安全办 第 4 名员工是 李冰冰
==========================================
质量安全办 共有 4 名员工
------------------------------------------
质量安全办 第 0 名员工是 特朗普
质量安全办 第 1 名员工是 拜登
质量安全办 第 2 名员工是 邓文迪
质量安全办 第 3 名员工是 李冰冰

这是未完全实现的版本。