思路
1、方法1:递归
每次将left与right中间的值nums[center]填入当前节点val中,再将nums数组切分为[left, center-1],[center+1, right],分别对左右子树再调用一次该函数,最后返回root
下面给出了递归代码
2、方法2:迭代
递归是可以用 迭代+栈 代替的,其实就是模拟递归,有兴趣的朋友可以写一下
举个例子:
[-10,-3,0,5,9]
第一步:创建两个栈
第二步:left=0,right=4
第三步:left<=right,创建root节点
第四步:将中间节点index = 2 存于root->val
第五步:将root压入第一个栈,4压入第二个栈
第六步:left=left=0,right=index-1=1
第七步:left<=right,创建l节点
第八步:将中间节点index = 0 存于l->val(l节点就是root的左子节点)
第九步:将 l 压入第一个栈,1压入第二个栈
第十步:left=left=0,right=index-1=-1
第十一步:left>right,弹栈,r=pop(第一个栈) = l,right=pop(第二个栈) = 1
第十二步:令left = l->val+1 = 1,此时right = 1
第十三步:left<=right,创建 r 节点
第十四步:将中间节点index = 1 存于r->val
第十五步:将 r 压入第一个栈,1压入第二个栈
第十六步:left=index+1=2,right=1
第十七步:left>right,弹栈,r=pop(第一个栈) = root,right=pop(第二个栈) = 4
此时root的左子树已建好,右子树也是按照上面的步骤进行建立
现在每个节点的val存储的都是索引,到最后左右子树建立完成时,遍历左右子树将索引值填入即可
代码
struct TreeNode* recursion(struct TreeNode *root, int *nums, int left, int right){
if(left > right) return NULL;
int center = (left + right) / 2;
root = (struct TreeNode*)malloc(sizeof(struct TreeNode));
root->val = nums[center];
root->left = recursion(root->left, nums, left, center-1);
root->right = recursion(root->right, nums, center+1, right);
return root;
}
struct TreeNode* sortedArrayToBST(int* nums, int numsSize){
struct TreeNode *ret;
ret = recursion(ret, nums, 0, numsSize-1);
return ret;
}