JAVA 快速构建树形结构

 

• JDK 1.8+
• Node 中pid 为 0 的是根节点

  @SpringBootTest
  @RunWith(SpringRunner.class)
  public class Test {
          private static final Logger log = LoggerFactory.getLogger(Test.class);
          Node dennis = new Node(1, 0, "dennis");
          Node calm = new Node(2, 0, "calm");
          Node daughter = new Node(3, 1, "daughter");
          Node grandson = new Node(4, 3, "grandson");
          Node son = new Node(5, 2, "son");
          List<Node> nodes = Lists.newArrayList(dennis, calm, daughter, son, grandson);
          List<Node> tree = TreeUtil.buildTree(nodes);
          log.info("tree is create: {}", JSON.toJSONString(tree));
  }

  public class Node {
      private int id;
      private int pid;
      private String name;
      private List<Node> sub = new ArrayList<>();
      public Node(int id, int pid) {
          this.id = id;
          this.pid = pid;
      }
  
      public Node(int id, int pid, String name) {
          this(id, pid);
          this.name = name;
      }
  
      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;
      }
  
      public List<Node> getSub() {
          return sub;
      }
  
      public void setSub(List<Node> sub) {
          this.sub = sub;
      }
  }

 

[
  {
    "id": 1,
    "name": "dennis",
    "pid": 0,
    "sub": [
      {
        "id": 3,
        "name": "daughter",
        "pid": 1,
        "sub": [
          {
            "id": 4,
            "name": "grandson",
            "pid": 3
          }
        ]
      }
    ]
  },
  {
    "id": 2,
    "name": "calm",
    "pid": 0,
    "sub": [
      {
        "id": 5,
        "name": "son",
        "pid": 2
      }
    ]
  }
]