Java的树形结构遍历

定义 Menu

import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.Accessors;

import java.util.List;

/**
 * @ClassName Menu
 * @Description TODO
 * @Author YZJ
 * @Date 2022/11/27 20:31
 **/
@Data
@AllArgsConstructor
@NoArgsConstructor
@Accessors(chain = true)
public class Menu {

    private Long id;
    //菜单名称
    private String menuName;
    //父菜单ID
    private Long parentId;

    private List<Menu> children;
}

生成树形结构

public static void main(String[] args) {
        Menu menu = new Menu(0L, "国家", -1L, null);
        Menu menu1 = new Menu(1L, "中国", 0L, null);
        Menu menu2 = new Menu(2L, "日本", 0L, null);
        Menu menu3 = new Menu(3L, "美国", 0L, null);

        Menu menu4 = new Menu(4L, "北京", 1L, null);
        Menu menu5 = new Menu(5L, "上海", 1L, null);
        Menu menu6 = new Menu(6L, "武汉", 1L, null);

        Menu menu7 = new Menu(7L, "东京", 2L, null);
        Menu menu8 = new Menu(8L, "大阪", 2L, null);

        Menu menu9 = new Menu(9L, "华盛顿", 3L, null);
        Menu menu10 = new Menu(10L, "纽约", 3L, null);
        Menu menu11 = new Menu(11L, "加州", 3L, null);

        Menu menu12 = new Menu(12L, "洪山区", 6L, null);
        Menu menu13 = new Menu(13L, "武昌区", 6L, null);
        Menu menu14 = new Menu(14L, "汉阳区", 6L, null);

        List<Menu> menus = new ArrayList<>();
        menus.add(menu);
        menus.add(menu1);
        menus.add(menu2);
        menus.add(menu3);
        menus.add(menu4);
        menus.add(menu5);
        menus.add(menu6);
        menus.add(menu7);
        menus.add(menu8);
        menus.add(menu9);
        menus.add(menu10);
        menus.add(menu11);
        menus.add(menu12);
        menus.add(menu13);
        menus.add(menu14);

        // 从 0 开始构建整个树
        long parentId = 0L;

        System.out.println(JSON.toJSONString(builderMenuTree(menus, parentId)));
    }
    private static List<Menu> builderMenuTree(List<Menu> menus, long parentId) {
        return menus.stream()
                .filter(menu -> menu.getParentId() != null && menu.getParentId().equals(parentId))
                .map(menu -> menu.setChildren(getChildren(menu, menus)))
                .collect(Collectors.toList());
    }

    private static List<Menu> getChildren(Menu menu, List<Menu> menus) {
        return menus.stream()
                .filter(m ->  menu.getParentId() != null && m.getParentId().equals(menu.getId()))
                .map(m -> m.setChildren(getChildren(m,menus)))
                .collect(Collectors.toList());
    }

输出

[
    {
        "children":[
            {
                "children":[

                ],
                "id":4,
                "menuName":"北京",
                "parentId":1
            },
            {
                "children":[

                ],
                "id":5,
                "menuName":"上海",
                "parentId":1
            },
            Object{...}
        ],
        "id":1,
        "menuName":"中国",
        "parentId":0
    },
    {
        "children":Array[2],
        "id":2,
        "menuName":"日本",
        "parentId":0
    },
    {
        "children":Array[3],
        "id":3,
        "menuName":"美国",
        "parentId":0
    }
]

  • 0
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值