java递归获取某个父节点下面的所有子节点

package top.atpisher.com.tacocloud.test;

public class TreeNode{
    int id;
    int pid;
    String name;

    public TreeNode() {
        super();
    }
    public TreeNode(int id, int pid) {
        super();
        this.id = id;
        this.pid = pid;
    }
    public int getId() {
        return id;
    }
    public void setId(int id) {
        this.id = id;
    }
    public int getPid() {
        return pid;
    }
    public void setPid(int pid) {
        this.pid = pid;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
}
package top.atpisher.com.tacocloud.test;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class TreeUtil {

    //子节点
    static List<TreeNode> childMenu = new ArrayList<TreeNode>();
    public  List<TreeNode> treeMenuList1(List<TreeNode> menuList, int pid) {
        for (TreeNode mu : menuList) {
            //遍历出父id等于参数的id,add进子节点集合
            if (Integer.valueOf(mu.getPid()) == pid) {
                //递归遍历下一级
                treeMenuList1(menuList, Integer.valueOf(mu.getId()));
                childMenu.add(mu);
            }
        }
        return childMenu;
    }
    // 过滤掉未选中的记录
    public  List<TreeNode> fiterCatalogs(List<TreeNode> allData, Integer selectid) {
        List<TreeNode> chooseData = new ArrayList<>();
        Queue<Integer> query = new LinkedList<Integer>();
        query.add(selectid);
        for (TreeNode item : allData) {
            if (selectid.equals(item.getId())) {
                chooseData.add(item);
                break;
            }
        }
        while (!query.isEmpty()) {
            Integer selectidTemp = query.poll();
            for (TreeNode item : allData) {
                if (selectidTemp.equals(item.getPid())) {
                    query.add(item.getId());
                    chooseData.add(item);
                }
            }
        }

        return chooseData;
    }

    /**
     * 0---1----4
     *     +---5
     * +---2---6
     *     +---7
     * +---3---8---15
     *     +---9---10---11
     *             +---12
     *             +---13
     *              +---14
     * @param args
     */
    public static void main(String[] args) {
        TreeUtil tree=new TreeUtil();
        List<TreeNode>  treeNodes = new  ArrayList<TreeNode>();
        List<TreeNode>  childNode = new  ArrayList<TreeNode>();
        treeNodes.add(new TreeNode(1,0));
        treeNodes.add(new TreeNode(2,0));
        treeNodes.add(new TreeNode(3,0));

        treeNodes.add(new TreeNode(4,1));
        treeNodes.add(new TreeNode(5,1));

        treeNodes.add(new TreeNode(6,2));
        treeNodes.add(new TreeNode(7,2));

        treeNodes.add(new TreeNode(8,3));
        treeNodes.add(new TreeNode(9,3));

        treeNodes.add(new TreeNode(15,8));
        treeNodes.add(new TreeNode(10,9));
        treeNodes.add(new TreeNode(11,10));
        treeNodes.add(new TreeNode(12,10));
        treeNodes.add(new TreeNode(13,10));
        treeNodes.add(new TreeNode(14,10));

        int rootId = 0;
        System.out.println("查找 rootId = " + rootId + "的子节点");
        // 开始时间
        long stime = System.nanoTime();
        tree.treeMenuList1(treeNodes,rootId);
//        for(TreeNode node :) {
//            System.out.println(" childNode id = " + node.getId());
//        }
        // 结束时间
        long etime = System.nanoTime();
        // 计算执行时间
        System.out.printf("执行时长:%d 纳秒.", (etime - stime));
        System.out.println();

        // 开始时间
        long stime1 = System.nanoTime();
        tree.fiterCatalogs(treeNodes, rootId);
//        for(TreeNode node : ) {
//            System.out.println(" childNode id = " + node.getId());
//        }
        // 结束时间
        long etime1 = System.nanoTime();
        // 计算执行时间
        System.out.printf("执行时长:%d 纳秒.", (etime1 - stime1));
    }
}

运行结果

查找 rootId = 0的子节点
执行时长:376200 纳秒.
执行时长:116700 纳秒.
Process finished with exit code 0

求助大佬,能不能有效率高一点的方法

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值