list 转成 tree

package com.zl;

import java.util.ArrayList;
import java.util.List;

public class MenuItem {
    private String id;
    private String pid;
    private String name;
    private List<MenuItem> childMenuItemList;

    public List<MenuItem> getChildMenuItemList() {
        return childMenuItemList;
    }

    public void setChildMenuItemList(List<MenuItem> childMenuItemList) {
        this.childMenuItemList = childMenuItemList;
    }

    public String getId() {
        return id;
    }

    public void setId(String id) {
        this.id = id;
    }

    public String getPid() {
        return pid;
    }

    public void setPid(String pid) {
        this.pid = pid;
    }

    public String getName() {
        return name;
    }

    public void setName(String name) {
        this.name = name;
    }

    @Override
    public String toString() {
        return "MenuItem{" +
                "id='" + id + '\'' +
                ", pid='" + pid + '\'' +
                ", name='" + name + '\'' +
                ", childMenuItemList=" + childMenuItemList +
                '}';
    }
}

　　

package com.zl;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * Hello world!
 */
public class App {
    /**
     * 把 list 转化成 tree 的形式
     * 核心是 对象都是引用的,对象间引用添加关联
     * @param itemList
     */
    public static Map<String, MenuItem> listToTree(List<MenuItem> itemList) {
        //存放树结构的map,返回数据
        HashMap<String, MenuItem> resultMap = new HashMap<>();
        //map初始化 key为id,value为对象,方便根据id直接获取对象
        HashMap<String, MenuItem> menuItemHashMap = new HashMap<>();
        for (MenuItem menuItem : itemList) {
            menuItemHashMap.put(menuItem.getId(), menuItem);
        }
        for (MenuItem menuItem : itemList) {
            String pid = menuItem.getPid();
            if (pid != null && pid != "") {
                MenuItem parentMenuItem = menuItemHashMap.get(pid);
                //有父节点的把当前节点挂在父节点上,没有父节点的说明当前节点就是根节点
                if (null != parentMenuItem) {
                    List<MenuItem> childMenuItemList = parentMenuItem.getChildMenuItemList();
                    if (null != childMenuItemList) {
                        childMenuItemList.add(menuItem);

                    } else {
                        childMenuItemList = new ArrayList<>();
                        parentMenuItem.setChildMenuItemList(childMenuItemList);
                        childMenuItemList.add(menuItem);
                    }
                }else {
                    //没有找到父节点说明当前节点是根节点,放到返回的map中
                    resultMap.put(menuItem.getId(),menuItem);
                }
            }else {
                //没有找到父节点说明当前节点是根节点,放到返回的map中
                resultMap.put(menuItem.getId(),menuItem);
            }

        }
        System.out.println(resultMap);
        return resultMap;
    }

    public static void main(String[] args) {
        ArrayList<MenuItem> menuItemArrayList = new ArrayList<>();
        MenuItem menuItem = new MenuItem();
        MenuItem menuItem2 = new MenuItem();
        MenuItem menuItem3 = new MenuItem();
        MenuItem menuItem4 = new MenuItem();
        MenuItem menuItem5 = new MenuItem();
        MenuItem menuItem6 = new MenuItem();
        MenuItem menuItem7 = new MenuItem();
        MenuItem menuItem8 = new MenuItem();

        menuItemArrayList.add(menuItem);
        menuItemArrayList.add(menuItem2);
        menuItemArrayList.add(menuItem3);
        menuItemArrayList.add(menuItem4);
        menuItemArrayList.add(menuItem5);
        menuItemArrayList.add(menuItem6);
        menuItemArrayList.add(menuItem7);
        menuItemArrayList.add(menuItem8);

        menuItem.setId("1");
        menuItem.setPid("0");
        menuItem.setName("1");

        menuItem2.setId("2");
        menuItem2.setPid("0");
        menuItem2.setName("2");

        menuItem3.setId("3");
        menuItem3.setPid("2");
        menuItem3.setName("2.1");

        menuItem4.setId("4");
        menuItem4.setPid("0");
        menuItem4.setName("4");

        menuItem5.setId("5");
        menuItem5.setPid("4");
        menuItem5.setName("4.1");

        menuItem6.setId("6");
        menuItem6.setPid("5");
        menuItem6.setName("4.1.1");

        menuItem7.setId("7");
        menuItem7.setPid("6");
        menuItem7.setName("4.1.1.1");

        menuItem8.setId("8");
        menuItem8.setPid("6");
        menuItem8.setName("4.1.1.2");

        listToTree(menuItemArrayList);
    }
}