leetcode 108. 将有序数组转换为二叉搜索树
给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵
平衡二叉搜索树。
解题思路:
直观地看,我们可以选择中间数字作为二叉搜索树的根节点,这样分给左右子树的数字个数相同或只相差 1,可以使得树保持平衡。如果数组长度是奇数,则根节点的选择是唯一的,如果数组长度是偶数,则可以选择中间位置左边的数字作为根节点或者选择中间位置右边的数字作为根节点,选择不同的数字作为根节点则创建的平衡二叉搜索树也是不同的。
确定平衡二叉搜索树的根节点之后,其余的数字分别位于平衡二叉搜索树的左子树和右子树中,左子树和右子树分别也是平衡二叉搜索树,因此可以通过递归的方式创建平衡二叉搜索树。
递归的基准情形是平衡二叉搜索树不包含任何数字,此时平衡二叉搜索树为空。
#include <stdio.h>
#include <stdlib.h>
struct TreeNode {int val;struct TreeNode *left;struct TreeNode *right;
};
//选定中间位置的数字作为根节点
struct TreeNode *help(int *nums,int left,int right)
{if(left>right)return NULL;//选择中间位置的数字作为根节点int mid=(left+right)/2;//创建根节点struct TreeNode *root;root=malloc(sizeof(struct TreeNode));root->val=nums[mid];root->left=help(nums,left,mid-1);root->right=help(nums,mid+1,right);return root;
}
struct TreeNode* sortedArrayToBST(int* nums, int numsSize) {return help(nums,0,numsSize-1);
}
// 层次遍历输出
void printLevelOrder(struct TreeNode* root) {if (root == NULL) return;// 使用队列来存储节点struct TreeNode** queue = malloc(100 * sizeof(struct TreeNode*)); // 假设最大层数为100int front = 0, rear = 0;// 将根节点加入队列queue[rear++] = root;while (front < rear) {struct TreeNode* current = queue[front++];printf("%d ", current->val); // 打印当前节点的值// 如果左子树存在,加入队列if (current->left != NULL) {queue[rear++] = current->left;}// 如果右子树存在,加入队列if (current->right != NULL) {queue[rear++] = current->right;}}free(queue); // 释放队列内存
}int main()
{struct TreeNode* root=NULL; // int nums[] ={-10,-3,0,5,9}; [1,3]int nums[] ={1,3};root=sortedArrayToBST(nums,sizeof(nums)/sizeof(nums[0]));if(root==NULL)return -1;printLevelOrder(root);return 0;
}