Golang樹結構的實現

Golang樹結構的實現。

 

樹結構Golang源代碼：

 

/*
**
** Version : 0.1
**
 */

package mnds

import (
    "errors"
    "fmt"
    "strings"
)

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

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

func (this *Tree) GetName() string {
    // Bug: when this is root, can get an nil parent point, it will panic...
    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) GetParent() *Tree {
    return this.parent
}

func (this *Tree) GetPath() []*Tree {
    var path []*Tree

    path = append(path, this)
    p := this.parent
    if p != nil {
        path = append(path, p.GetPath()...)
    }

    return path
}

func (this *Tree) AddSon(son interface{}) {
    switch son.(type) {
    case string:
        var Son Tree
        Son.name = son.(string)
        Son.parent = this
        this.son = append(this.son, &Son)
    case *Tree:
        this.son = append(this.son, son.(*Tree))
        son.(*Tree).parent = this
    default:
        break
    }
}

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) FindNode(nameofnode string) []*Tree {
    var t []*Tree

    if strings.Contains(this.GetName(), nameofnode) {
        t = append(t, this)
    } else {
        if this.son == nil {
            t = nil
        } else {
            for _, v := range this.son {
                t = append(t, v.FindNode(nameofnode)...)
            }
        }
    }

    return t
}

func (this Tree) GetDegree() int {
    degree := len(this.son)

    if this.son != nil {
        for _, v := range this.son {
            if degree < v.GetDegree() {
                degree = v.GetDegree()
            }
        }
    }

    return degree
}

func (this Tree) GetDepth() int {
    var depth int
    p := this.parent
    if p == nil {
        depth = 0
    } else {
        depth = p.GetDepth() + 1
    }

    return depth
}

func (this Tree) GetHeight() int {
    var height int = 0

    if this.son == nil {
        height = 0
    } else {
        sonheight := make([]int, len(this.son))
        // Get all height of sons
        for k, v := range this.son {
            if v.son != nil {
                // If have son, get its height, and must add 1
                sonheight[k] = v.GetHeight() + 1
            } else {
                // If son havn't son, and his parent's height is 1
                height = 1
            }
        }

        // Get the max height
        for _, v := range sonheight {
            if height < v {
                height = v
            }
        }
    }

    return height
}

// To show all data of the tree.
// You can get all data of the tree like this.
func (this Tree) ShowAll() {
    for k := 0; k < this.GetDepth(); k++ {
        fmt.Printf("  ")
    }
    fmt.Printf("%s\n", this.GetName())
    if this.son != nil {
        for _, v := range this.son {
            for k := 0; k < v.GetDepth(); k++ {
                fmt.Printf("  ")
            }
            v.ShowAll()
        }
    }
}

func (this Tree) ToJson() []byte {

    return nil
}

func (this *Tree) FromJson(data []byte) error {

    return nil
}

func (this Tree) ToMap() map[string]string {

    return nil
}

func (this *Tree) FromMap(data map[string]string) error {

    return nil
}

 

測試代碼：

 

package main

import (
    "fmt"

    "merrynuts/mnds"
)

type country struct {
    mnds.Tree
    countryname string `json:"countryname"`
}

func main() {
    var china country
    var shanghai mnds.Tree
    var leliu mnds.Tree

    china.Create("中國")
    china.countryname = china.GetName()
    fmt.Println(china.countryname)

    china.AddSon("廣東省")
    china.AddSon("吉林省")
    china.AddSon("廣西壯族自治區")
    china.AddSon("湖南省")
    china.AddSon("湖北省")

    hn, err := china.GetSon("湖南省")
    if err != nil {
        fmt.Println(china.GetName(), "沒有一個叫湖南省的兒子")
    } else {
        hn.AddSon("長沙市")
        hn.AddSon("衡陽市")
        hn.AddSon("張家界市")
        cs, err := hn.GetSon("長沙市")
        if err != nil {
            fmt.Println(cs.GetName(), "沒有一個叫長沙市的兒子")
        } else {
            cs.AddSon("芙蓉區")
            cs.AddSon("天心區")
            cs.AddSon("岳麓區")
            cs.AddSon("開福區")
            cs.AddSon("雨花區")
            cs.AddSon("長沙縣")
            cs.AddSon("望城區")
            cs.AddSon("瀏陽市")
            cs.AddSon("寧鄉市")
        }
    }

    gd, err := china.GetSon("廣東省")
    if err != nil {
        fmt.Println(china.GetName(), "沒有一個叫廣東省的兒子")
    } else {
        gd.AddSon("深圳市")
        gd.AddSon("廣州市")
        gd.AddSon("佛山市")
        gd.AddSon("中山市")
        gd.AddSon("東莞市")
        fs, err := gd.GetSon("佛山市")
        if err != nil {
            fmt.Println(gd.GetName(), "沒有一個叫佛山市的兒子")
        } else {
            fs.AddSon("順德區")
            fs.AddSon("南海區")
            fs.AddSon("禪城區")
            fs.AddSon("三水區")
            fs.AddSon("高明區")
            sd, err := fs.GetSon("順德區")
            if err != nil {
                fmt.Println(fs.GetName(), "沒有一個叫順德區的兒子")
            } else {
                sd.AddSon("大良街道")
                sd.AddSon("龍江街道")
                sd.AddSon("倫教街道")
                sd.AddSon("樂從街道")
                sd.AddSon("容桂街道")
                sd.AddSon("杏壇街道")
                sd.AddSon("均安街道")

                leliu.Create("勒流街道")
                sd.AddSon(&leliu)
            }

            nh, err := fs.GetSon("南海區")
            if err != nil {
                fmt.Println(fs.GetName(), "沒有一個叫南海區的兒子")
            } else {
                nh.AddSon("獅山街道")
                nh.AddSon("羅村街道")
                nh.AddSon("丹灶街道")
            }
        }
    }

    gx, err := china.GetSon("廣西壯族自治區")
    if err != nil {
        fmt.Println(china.GetName(), "沒有一個叫廣西壯族自治區的兒子")
    } else {
        gx.AddSon("桂林市")
        gx.AddSon("柳州市")
        gx.AddSon("梧州市")
        gx.AddSon("北海市")
        gl, err := gx.GetSon("桂林市")
        if err != nil {
            fmt.Println(gx.GetName(), "沒有一個叫桂林市的兒子")
        } else {
            gl.AddSon("桂平市")
            gl.AddSon("順德區")
        }
        var hechen mnds.Tree
        hechen.Create("河辰市")
        gx.AddSon(&hechen)
    }

    shanghai.Create("上海市")
    shanghai.AddSon("靜安區")
    shanghai.AddSon("浦東區")
    china.AddSon(&shanghai)

    fmt.Println(china.GetName(), "的整體結構如下：")
    fmt.Println("===========================================================")
    china.ShowAll()
    fmt.Println("-----------------------------------------------------------")

    for {
        var ss string
        fmt.Printf("Search? ")
        fmt.Scanf("%s\n", &ss)
        if ss == "exit" {
            break
        }
        result := china.FindNode(ss)
        if result == nil {
            fmt.Println("Can't find the node name's", ss)
        } else {
            for k, v := range result {
                fmt.Println("===========================================================")
                fmt.Println("Find the no.", k+1 , "of node name's", ss, "It's path to root is :")
                path := v.GetPath()
                for k := len(path) - 1; k >= 0; k-- {
                    if k > 0 {
                        fmt.Printf("%s-->", path[k].GetName())
                    } else {
                        fmt.Printf("%s", path[k].GetName())
                    }
                }
                fmt.Println()
                fmt.Println("It's structure is :")
                v.ShowAll()
                fmt.Println("-----------------------------------------------------------")
            }
        }
    }
}

 

 

運行效果：

 

D:\project\merrynuts\src>merrynuts
中國
中國 的整體結構如下：
===========================================================
中國
    廣東省
        深圳市
        廣州市
        佛山市
            順德區
                大良街道
                龍江街道
                倫教街道
                樂從街道
                容桂街道
                杏壇街道
                均安街道
                勒流街道
            南海區
                獅山街道
                羅村街道
                丹灶街道
            禪城區
            三水區
            高明區
        中山市
        東莞市
    吉林省
    廣西壯族自治區
        桂林市
            桂平市
            順德區
        柳州市
        梧州市
        北海市
        河辰市
    湖南省
        長沙市
            芙蓉區
            天心區
            岳麓區
            開福區
            雨花區
            長沙縣
            望城區
            瀏陽市
            寧鄉市
        衡陽市
        張家界市
    湖北省
    上海市
        靜安區
        浦東區
-----------------------------------------------------------
Search?