好的,下面是一个使用 Java 生成树形结构的代码示例。首先,我们可以定义一个节点类,用来表示树中的每个节点:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val) {
this.val = val;
this.left = null;
this.right = null;
}
}
然后,我们可以定义一个生成树形结构的函数,比如叫做 buildTree
,它接受两个参数,一个是节点值的数组,一个是节点的父节点的索引的数组,返回值是根节点。
public TreeNode buildTree(int[] values, int[] parentIndices) {
// 根据节点值数组和父节点索引数组,创建节点列表
List<TreeNode> nodes = new ArrayList<>();
for (int i = 0; i < values.length; i++) {
nodes.add(new TreeNode(values[i]));
}
// 遍历节点列表,为每个节点设置左右子节点
for (int i = 0; i < nodes.size(); i++) {
TreeNode node = nodes.get(i);
int leftIndex = 2 * i + 1;
int rightIndex = 2 * i + 2;
if (leftIndex < nodes.size()) {
node.left = nodes.get(leftIndex);
}
if (rightIndex < nodes.size()) {
node.right = nodes.get(rightIndex);
}
}
// 返回根节点
return nodes.get(0);
}
该函数使用一个节点值的数组和一个父节点的索引的数组,来创建一个节点的列表,然后遍历该列表,为每