leetcode 1008. 先序遍历构造二叉树 java
返回与给定先序遍历 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 中的值是不同的。
思路:
新节点找到位置,插入,回退到根节点
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { = x; }
* }
*/
class Solution {
public TreeNode bstFromPreorder(int[] preorder) {
TreeNode root = null;
for (int i = 0; i < preorder.length; i++) {
System.out.println(preorder[i]);
root = add(root, preorder[i]);
}
return root;
}
public TreeNode add(TreeNode root, int num) {
if (root == null) {
System.out.println(1);
return new TreeNode(num);
}
if (root.val > num) {
System.out.println(2);
root.left = add(root.left, num);
} else {
System.out.println(3);
root.right = add(root.right, num);
}
System.out.println(4);
//回退到根节点
return root;
}
}
思路2:
找到指定位置,划分左右子树
class Solution {
public TreeNode bstFromPreorder(int[] preorder) {
if (preorder == null){
return null;
}
return bfp(preorder, 0, preorder.length - 1);
}
//找到第一个比根节点大的,来划分左右子树
TreeNode bfp(int[] preorder, int ls, int le) {
if (ls > le)
return null;
int cur = preorder[ls];
TreeNode curNode = new TreeNode(cur);
int index = le + 1;
for (int i = ls + 1; i <= le; i++) {
if (preorder[i] > cur) {
index = i;
break;
}
}
curNode.left = bfp(preorder, ls + 1, index - 1);
curNode.right = index > le ? null : bfp(preorder, index, le);
return curNode;
}
}