leetcode 108.将有序数组转换为二叉搜索树(convert sorted array to binary search tree)C语言
1.description
https://leetcode-cn.com/problems/convert-sorted-array-to-binary-search-tree/
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例:
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
2.solution
常规版本
struct TreeNode *addNode(int *nums, int left, int right){
if(left>right){
return NULL;
}
struct TreeNode *node = (struct TreeNode*)malloc(sizeof(struct TreeNode));
int mid = left + (right - left) / 2;
node->val = nums[mid];
node->left = addNode(nums, left, mid-1);
node->right = addNode(nums, mid+1, right);
return node;
}
struct TreeNode* sortedArrayToBST(int* nums, int numsSize){
return addNode(nums, 0, numsSize-1);
}
不需要辅助函数:
struct TreeNode* sortedArrayToBST(int* nums, int numsSize){
// 这种写法不需要辅助函数
// numsSize==0说明本次递归不会有节点产生,即递归结束
if(numsSize==0) return NULL;
// mid为当前根节点的索引
int mid = numsSize/2;
struct TreeNode *root = (struct TreeNode*)malloc(sizeof(struct TreeNode));
root->val = nums[mid];
root->left = sortedArrayToBST(nums, mid);
// 通过移位nums指针、改变numsSize大小缩小了数组范围,这样下一个根节点一定在"新数组"最中间
// 注意numsSize-mid-1不要写成mid-1,两者只有在numsSize为偶数时才相等,奇数则不相等
root->right = sortedArrayToBST(nums+mid+1, numsSize-mid-1);
return root;
}