go实现树形叶子节点选取

今天项目增加功能的时候用到了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
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Devil枫

发财小手鼓励一下作者大大

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值