目标
给你一个二维整数数组 descriptions ,其中 descriptions[i] = [parenti, childi, isLefti] 表示 parenti 是 childi 在 二叉树 中的 父节点,二叉树中各节点的值 互不相同 。此外:
- 如果 isLefti == 1 ,那么 childi 就是 parenti 的左子节点。
- 如果 isLefti == 0 ,那么 childi 就是 parenti 的右子节点。
请你根据 descriptions 的描述来构造二叉树并返回其 根节点 。
测试用例会保证可以构造出 有效 的二叉树。
示例 1:
输入:descriptions = [[20,15,1],[20,17,0],[50,20,1],[50,80,0],[80,19,1]]
输出:[50,20,80,15,17,19]
解释:根节点是值为 50 的节点,因为它没有父节点。
结果二叉树如上图所示。
示例 2:
输入:descriptions = [[1,2,1],[2,3,0],[3,4,1]]
输出:[1,2,null,null,3,4]
解释:根节点是值为 1 的节点,因为它没有父节点。
结果二叉树如上图所示。
说明:
- 1 <= descriptions.length <= 10^4
- descriptions[i].length == 3
- 1 <= parenti, childi <= 10^5
- 0 <= isLefti <= 1
- descriptions 所描述的二叉树是一棵有效二叉树
思路
有一个二维数组 descriptions,descriptions[i] = [parenti, childi, left or right] 描述了二叉树中的一对父子关系,即 parenti 是 childi 的父节点,childi 是 parenti 的左或者右孩子(取决于 1 或者 0)。重建二叉树并返回根节点。题目保证描述的是有效二叉树,且节点值不重复。
使用哈希表保存树节点,根据描述关系将节点连接起来,最终需要找出根节点,可以将孩子节点存到哈希集合中,返回不在该集合的节点即可。
代码
/**
* @date 2026-06-08 11:34
*/
public class CreateBinaryTree2196 {
public TreeNode createBinaryTree(int[][] descriptions) {
Map<Integer, TreeNode> map = new HashMap<>();
Set<Integer> childKeySet = new HashSet<>();
for (int[] description : descriptions) {
int parentKey = description[0];
map.putIfAbsent(parentKey, new TreeNode(parentKey));
TreeNode parent = map.get(parentKey);
int childKey = description[1];
childKeySet.add(childKey);
map.putIfAbsent(childKey, new TreeNode(childKey));
TreeNode child = map.get(childKey);
if (description[2] == 1) {
parent.left = child;
} else {
parent.right = child;
}
}
for (int[] description : descriptions) {
int parentKey = description[0];
if (!childKeySet.contains(parentKey)) {
return map.get(parentKey);
}
}
return null;
}
}
性能
