题目:
Given an array where elements are sorted in ascending order, convert it to a height balanced BST.
把一个已经排好序(升序)的数组,转化为一个平衡二叉搜索树。
思路:
本题需要先了解二叉搜索树(参看:http://blog.csdn.net/u012243115/article/details/40779429)和平衡二叉树。
了解了之后,可以发现平衡二叉搜索树的根结点的值正好是所有节点值的中间值,即中序遍历时,根结点位于正中间(因为是平滑二叉树)(如果是偶数个结点,则位于最中间两个靠前的那个)。于是可以想到使用二分法,先把有序数组的最中间值设为根结点,然后根结点的左孩子是左半段数组的最中间值,根结点的右孩子是右半段数组的最中间值。依次这样,即可以建立一个平衡二叉搜索树。
代码:
/**
* Definition for binary tree
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
TreeNode *sortedArrayToBST(vector<int> &num)
{
return sortedArrayToBST(num , 0 , num.size() - 1);
}
//使用二分法建立平衡二叉搜索树
TreeNode *sortedArrayToBST(vector<int> &num , int begin , int end)
{
//递归结束条件
if(begin > end)
return NULL;
int mid = (begin + end)/2;
TreeNode *root = new TreeNode(num[mid]);//根结点,为当前数组的中间值
root->left = sortedArrayToBST(num , begin , mid - 1);//根结点的左孩子为左半段数组的中间值
root->right = sortedArrayToBST(num , mid + 1 , end);//根结点的右孩子为右半段数组的中间值
return root;
}
};