先序遍历构造二叉树(Construct Binary Search Tree from Preorder Traversal)java_leetcode 1008
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/zhangjingao/article/details/90379789
题干
返回与给定先序遍历 preorder 相匹配的二叉搜索树(binary search tree)的根结点。
(回想一下,二叉搜索树是二叉树的一种,其每个节点都满足以下规则,对于 node.left 的任何后代,值总 < node.val,而 node.right 的任何后代,值总 > node.val。此外,先序遍历首先显示节点的值,然后遍历 node.left,接着遍历 node.right。)
示例:
输入:[8,5,1,7,10,12]
输出:[8,5,10,1,7,null,12]
提示:
1 <= preorder.length <= 100
先序 preorder 中的值是不同的。
分析
这个题主要考察二叉搜索树的构造,二叉搜索树就是,当前节点的值大于左节点的所有节点的值,但是小于右节点的所有节点的值。本题目中给出了先序遍历,其实就是在给出一个根节点后,将各个节点插入到二叉搜索树中,从根节点一次往下遍历。
代码
public TreeNode bstFromPreorder(int[] preorder) {
if (preorder == null) {
return null;
}
TreeNode root = new TreeNode(preorder[0]);
for (int i = 1; i < preorder.length; i++) {
work(root, preorder[i]);
}
return root;
}
private void work(TreeNode root, int value) {
if (value > root.val) {
if (root.right == null) {
root.right = new TreeNode(value);
} else {
work(root.right, value);
}
} else {
if (root.left == null) {
root.left = new TreeNode(value);
} else {
work(root.left, value);
}
}
}