今天项目增加功能的时候用到了go后台来实现单位树形的叶子节点选取,记录一下。
基本思路
提到树形获取节点,作为程序员的第一反应,基本都是递归,我也不例外,毕竟树这种数据结构有很强的递归性,遍历树、查询高度等等操作其实都可以递归的方式来实现。
实现代码
获取单位本身和子节点
传入参数:节点id、树形数据
例如(1,[[1,0],[2,1],[3,2],[4,2]])
func GetOrganChild(organ int, organsTree [][2]int) []int{
ids :=[]int{organ}
for i := 0; i < len(organsTree); i++ {
c_id :=organsTree[i][1]
c_ids :=[]int{}
if organ == c_id{
c_ids =GetOrganChild(organsTree[i][0], organsTree)
if len(c_ids) != 0{
ids = append(ids,c_ids...)
}
}
}
return ids
}
只获取叶子节点
func GetOrganChildEnd(organ int, organsTree [][2]int) []int{
ids :=[]int{organ}
for i := 0; i < len(organsTree); i++ {
c_id :=organsTree[i][1]
c_ids :=[]int{}
if organ == c_id{
c_ids =GetOrganChildEnd(organsTree[i][0], organsTree)
if len(c_ids) != 0{
ids = append(ids,c_ids...)
}
}
}
if len(ids) >1{
ids =ids[1:]
}
return ids
}