Leetcode 108. 将有序数组转换为二叉搜索树
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
示例
题目解释:
如题所示,将数组转换位平衡二叉树,何为平衡二叉树?即左右子树高度值最大相差1.
解答方法
1. 由于要保存高度平衡,即树根的左右子树所包含的节点数目应基本相等,或相差1个数目。
2. 且由于BST的特性(左<根<右),所有我们可以构想树根节点是有序数组的中间节点。
3. 然后再寻找左右子树的根节点,如何寻找?与上一步一致,即从左半部分的有序数组中获取中间位置的元素作为左子树的根节点,右半部分的有序数组中获取中间位置的元素作为右子树的根节点。
4. 因此,使用递归是一种很好的解答方法。
代码
/**
* 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)
{
return arrayToBST(nums , 0 , nums.size() - 1);
}
//使用二分法以保持平衡
TreeNode *arrayToBST(vector<int> &nums , int begin , int end)
{
// 划分结束
if(begin > end)
return NULL;
int mid = (begin + end)/2;
TreeNode *root = new TreeNode(nums[mid]); //根结点,为当前数组的中间值
root->left = arrayToBST(nums , begin , mid - 1); //左孩子为左半段数组的中间值
root->right = arrayToBST(nums , mid + 1 , end); //右孩子为右半段数组的中间值
return root;
}
};