C语言练习-LeetCode1008前序遍历构造二叉搜索树

题目

给定一个整数数组，它表示BST(即 二叉搜索树 )的先序遍历 ，构造树并返回其根。

保证 对于给定的测试用例，总是有可能找到具有给定需求的二叉搜索树。

二叉搜索树 是一棵二叉树，其中每个节点， Node.left 的任何后代的值 严格小于 Node.val , Node.right 的任何后代的值 严格大于 Node.val。

二叉树的 前序遍历 首先显示节点的值，然后遍历Node.left，最后遍历Node.right。

示例1



输入： preorder = [8,5,1,7,10,12]输出： [8,5,10,1,7,null,12]

示例2

输入: preorder = [1,3]输出: [1,null,3]

提示

• 1 <= preorder.length <= 100
• 1 <= preorder[i] <= 10^8
• preorder 中的值 互不相同

思路（拆分左右子树）

数组是由二叉树先序遍历得到，所以第一个元素start一定是根节点。我们得到根结点后将数组分为左子树部分和右子树部分，根据二叉搜索树的特征，右子树的值都是大于根结点，在此我们只需要找到第一个大于根节点的值的位置i即可将数组分为左子树部分start + 1 ~ i 和右子树部分i ~ end （左闭右开）。

区分完左右子树后，我们可以通过先序遍历来构建我们的树，构建一个函数bstFormPreorderRange,作用是将数组[start,end)位置的元素构建成树。

代码实现

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
};
  
  //将数组start ~ end位置的元素构建成树
struct TreeNode* bstFormPreorderRange(int* preorder,int start,int end) {
	//当start == end时，表示没有元素，返回NULL
	if (start == end) return NULL;
	
	//构建当前树的根结点，preorder[start]为根结点
	struct TreeNode* root = malloc(sizeof(struct TreeNode));
	root->val = preorder[start];
	//从start + 1向后寻找第一个值大于根结点的位置
	int i = start + 1;
	for (i = start + 1; i < end; i++) {
	    if (preorder[i] > root->val) {
	        break;
	    }
	}
	//通过递归分别构造左右子树
	root->left = bstFormPreorderRange(preorder, start + 1, i);
	root->right = bstFormPreorderRange(preorder,i , end);
	return root;
}
   
struct TreeNode* bstFromPreorder(int* preorder, int preorderSize) {
	return bstFormPreorderRange(preorder,0,preorderSize);
}