1、题目描述
将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。
本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。
2、示例
给定有序数组: [-10,-3,0,5,9],
一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:
0
/ \
-3 9
/ /
-10 5
3、题解
基本思想:递归,左右等分建立左右子树,中间节点作为子树根节点,不断递归次过程,小技巧传nums引用减小内存消耗。
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
void PreOrderTraverse(TreeNode* T)
{
if (T == NULL)
return;
cout << T->val << " ";
PreOrderTraverse(T->left);
PreOrderTraverse(T->right);
}
class Solution {
public:
TreeNode* sortedArrayToBST(vector<int>& nums) {
//基本思想:递归,左右等分建立左右子树,中间节点作为子树根节点,不断递归次过程,小技巧传nums引用减小内存消耗
TreeNode* root = NULL;
Recursion(nums, 0, nums.size() - 1, &root);
return root;
}
void Recursion(vector<int> &nums,int low,int high, TreeNode** root)
{
if (low > high)
return;
int mid = (low + high) / 2;
(*root) = new TreeNode(nums[mid]);
Recursion(nums, low, mid - 1, &(*root)->left);
Recursion(nums, mid + 1, high, &(*root)->right);
}
};
int main()
{
Solution solute;
vector<int> nums = { 1,2,3 };
TreeNode* root = solute.sortedArrayToBST(nums);
PreOrderTraverse(root);
return 0;
}