将扁平化数据菜单转为树结构（不使用递归）

今天药忘吃喽~ 2024-03-23 13:44 116阅读 0赞

### 将扁平化数据菜单转为树结构（不使用递归） ###

public class BaseControllerTest {
        
        public static void main(String[] args) throws Exception {
        
            List<Menu> data = new ArrayList<>();
            data.add(new Menu(1,5));
            data.add(new Menu(2,3));
            data.add(new Menu(3,5));
            data.add(new Menu(4,0));
            data.add(new Menu(5,0));
            data.add(new Menu(6,4));
            data.add(new Menu(7,5));
            Map<Integer, Menu> nodes = new HashMap<>();
            for (Menu flatNode : data) {
        
                nodes.put(flatNode.id, flatNode);
            }
    
    		/** 通过子id找父id唯一，如果通过父id找子id就不是唯一，
    		遍历所有节点，将其插入到其父节点的 children 列表中*/
            for (Menu flatNode : data) {
        
                Menu node = nodes.get(flatNode.id);
                Menu parent = nodes.get(flatNode.pid);
                if (parent != null) {
        
                    parent.children.add(node);
                }
            }
            List<Menu> root = new ArrayList<>();
            for (Menu node : nodes.values()) {
        
                if (node.pid == 0) {
        
                    root.add(node);
                }
            }
            System.out.println("");
        }
        static class Menu {
        
            public int id;
            public int pid;
            public List<Menu> children;
            public Menu(int id, int pid) {
        
                this.id = id;
                this.pid = pid;
                this.children = new ArrayList<>();
            }
        }
    
    }