java实现建树

最新推荐文章于 2024-08-08 02:25:29 发布

万卷读书破

最新推荐文章于 2024-08-08 02:25:29 发布

阅读量3.2k

点赞数

分类专栏： java基础文章标签：算法递归 java

本文链接：https://blog.csdn.net/ice_cream2588/article/details/82659175

版权

java基础专栏收录该内容

1 篇文章 0 订阅

订阅专栏

两层循环实现建树

创建节点类

public class TreeNode {
    protected int id;
    protected int parentId;
    protected List<TreeNode> children = new ArrayList<>();

    public void add(TreeNode node) {
        children.add(node);
    }
}

建树实现

/**
* 两层循环实现建树
*
* @param treeNodes 传入的树节点列表
* @return
*/
public static <T extends TreeNode> List<T> bulid(List<T> treeNodes, Object root) {

List<T> trees = new ArrayList<T>();

for (T treeNode : treeNodes) {

        if (root.equals(treeNode.getParentId())) {
            trees.add(treeNode);
        }

        for (T it : treeNodes) {
            if (it.getParentId() == treeNode.getId()) {
                if (treeNode.getChildren() == null) {
                    treeNode.setChildren(new ArrayList<TreeNode>());
                }
                treeNode.add(it);
            }
        }
    }
    return trees;
}

递归方法建树

创建节点类（同上）
建树实现

/**
 * 使用递归方法建树
 *
 * @param treeNodes
 * @return
 */
public static <T extends TreeNode> List<T> buildByRecursive(List<T> treeNodes, Object root) {
    List<T> trees = new ArrayList<T>();
    for (T treeNode : treeNodes) {
        if (root.equals(treeNode.getParentId())) {
            trees.add(findChildren(treeNode, treeNodes));
        }
    }
    return trees;
}