前言:
用数组生成一棵普通树,看到网上生成二叉树一般用递归资料较多,普通树没那个必要把,关于普通树的生成的资料比较匮乏,而且代码量普遍较大,很多没必要的代码,十分不简洁明了,本来以为java 自己有接口的,自己写了一个精简版的。
代码
package com.general;
import java.util.ArrayList;
import java.util.List;
/**
* @program: Work
* @description: 根据数组生成一棵普通树
* @author: LiuYing
* @create: 2019-03-19 11:05
**/
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class GenerateTree {
List<TreeNode> list = new ArrayList<>();
TreeNode root;
public GenerateTree(int[] arr) {
for (int i = 0; i < arr.length; i++) {
TreeNode node = new TreeNode(arr[i]);
list.add(node);
}
for (int i = 0; i < list.size() / 2; i++) {
if (2 * (i + 1) - 1 < list.size()) { //这个判断似乎是必须的
list.get(i).left = list.get(2 * (i + 1) - 1);
}
if (2 * (i + 1) < list.size()) {
list.get(i).right = list.get(2 * (i + 1));
}
}
root = list.get(0);
}
public static void main(String args[]) {
int a[] = {1, 2, 3, 4, 5, 6};
GenerateTree tree = new GenerateTree(a);
}
}