Convert Sorted Array to Binary Search Tree

最新推荐文章于 2022-02-19 22:56:38 发布

原创最新推荐文章于 2022-02-19 22:56:38 发布 · 161 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

42 篇文章

订阅专栏

本文介绍了一种将已排序的数组转换为高度平衡二叉搜索树的方法。平衡二叉树定义为任意节点的两个子树深度之差不超过1。通过递归地选取中间元素作为根节点来实现平衡。

PROBLEM：

Given an array where elements are sorted in ascending order, convert it to a height balanced BST.

For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.

Example:

Given the sorted array: [-10,-3,0,5,9],

One possible answer is: [0,-3,9,-10,null,5], which represents the following height balanced BST:

      0
     / \
   -3   9
   /   /
 -10  5

SOLVE：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    TreeNode* sortedArrayToBST(vector<int>& nums) {
        int head=0,rear=nums.size()-1;
        return toBST(nums,head,rear);
    }
    TreeNode* toBST(vector<int>& nums,int head,int rear){
        if(head>rear)
            return NULL;
        int middle=(rear+head+1)/2;
        TreeNode *node=new TreeNode(nums[middle]);
        node->left=toBST(nums,head,middle-1);
        node->right=toBST(nums,middle+1,rear);
        return node;
    }
};

注意：

1、结构体中定义了初始化函数，所以应该和类一样需要显示定义默认构造函数？然而此处没有，所以用“TreeNode *node=new TreeNode”会出现语法错误；

2、注意“int middle=(rear+head+1)/2”不能写成“int middle=(rear-head+1)/2”；

3、机构体中的成员初始化列表语法可以用于类中的const变量和普通变量（类中的const变量不能直接初始化），且只能用于构造函数中。