N3-1 - 数组 - convert-sorted-array-to-binary-search-tree

题目描述：

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

升序数组，转二叉排序树

解题思路：

使用递归方法：

选取中间的节点作为root节点，其左侧的节点构成左子树（递归调用建树），右侧的节点构成右子树。

class Solution {

public:

    TreeNode *sortedArrayToBST(vector<int> &num) {

        if(num.empty())

            return nullptr;//树为空

        return sortedArrayToBSTCore( num,0,num.size()-1);//对num取引用，避免了复制

    }

    TreeNode *sortedArrayToBSTCore(vector<int> &num, int begin,int end){

        //递归建立二叉搜索树

        if(end<begin)return nullptr;

        //if(end==begin)return num[begin];//只有一个元素时不再递归调用子树。返回类型错误应该是 TreeNode * 
        //只有一个元素时，不会调用if，不再递归，返回一个节点。因此不用特殊判断该情况

        int middle = (begin+end+1)/2; //取中间元素作为root节点,如果是偶数个元素，取中间偏后一个元素作为root节点

        TreeNode *pRoot = new TreeNode (num[middle]);

        if(middle>begin) //有左子树时

            pRoot->left = sortedArrayToBSTCore(num,begin, middle-1); //middle已经被使用

        if(middle<end)//有右子树

            pRoot->right = sortedArrayToBSTCore(num,middle+1, end);

        return pRoot;

    }

};

注意：

1. 测试序列中，root的选择，当子树节点为奇数时，取中间即可（begin+end）/2，当子树节点为偶数时，取中间偏后的节点（begin+end+1）/2。由于整数除法，向下取整，可统一式子为（begin+end+1）/2。错误的写法（begin+end）/2+1；在奇数个节点时，在取了正中间节点的后一位。

2. 除2，可以使用右移>>。（更好一些）

巴特西

N3-1 - 数组 - convert-sorted-array-to-binary-search-tree

题目描述：

解题思路：

最新文章

热门文章