链接:108. 将有序数组转换为二叉搜索树 - 力扣(LeetCode)
题目描述
给你一个整数数组 nums
,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡 二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
示例 1:
输入:nums = [-10,-3,0,5,9] 输出:[0,-3,9,-10,null,5] 解释:[0,-10,5,null,-3,null,9] 也将被视为正确答案:
示例 2:
输入:nums = [1,3] 输出:[3,1] 解释:[1,null,3] 和 [3,1] 都是高度平衡二叉搜索树。
题解
一、递归分治
由于数组是升序排列好的,需要构造高度平衡二叉树,所以将数组中间的值作为根节点,将左半部分作为左子树,右半部分作为右子树,这样就能保证构造的树是平衡的,以此类推,将前半部分数组和后半部分数组分别递归输入,通过递归就能得到一颗完整的高度平衡的二叉搜索树。
图示如下:
代码实现
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
TreeNode* sortedArrayToBST(vector<int>& nums) {
if(nums.size()==0){
return nullptr;
}
//调用排列函数
return ArrayToBST(nums,0,nums.size()-1);
}
TreeNode* ArrayToBST(vector<int>& nums,int start,int end){
if(start > end){
return nullptr;
}
//获取中间位置
int mid = (start + end)/2;
//中间节点构造为根节点
TreeNode* root = new TreeNode(nums[mid]);
//前半段作为左子树
root->left = ArrayToBST(nums,start,mid-1);
//后半段作为右子树
root->right = ArrayToBST(nums,mid+1,end);
//返回二叉树
return root;
}
};
总结
对二叉排序树的中序遍历结果是一个升序数组,因此,取升序数组任一位置的值作为根节点,左右两部分分别作为左右子树,就可以得到一个二叉排序树,而为了得到高度平衡的二叉树,就取靠近最中间的值作为根节点即可,当然选(low+high)/2,(low+high+1)/2 位置都是可行的。