给你一个整数数组 nums ,其中元素已经按 升序 排列,请你将其转换为一棵 高度平衡 二叉搜索树。
高度平衡二叉树:满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。
我的方法:先序遍历。由于数组已经排列好,只需取中间位置的元素作为根节点,递归操作,即可保证两边树的平衡(严谨来说需要证明)。
但是一开始我没想到,还以为要左旋右旋。其实这个题本质上跟二分查找思想一致。
构造新函数 helper,通过参数传递范围来帮助递归,似乎是关于搜索树的一种常见方法。之前做“给定先序和中序,生成后序”的时候,也是要用到这个技巧。
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode() : val(0), left(nullptr), right(nullptr) {}
TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
};
class Solution {
public:
TreeNode* helper(vector<int>& nums, int low, int high) {
if (low > high) return 0;
int mid = (high + low) / 2;
TreeNode* t = new TreeNode(nums[mid]);
t->left = helper(nums, low, mid - 1);
t->right = helper(nums, mid + 1, high);
return t;
}
TreeNode* sortedArrayToBST(vector<int>& nums) {
return helper(nums, 0, nums.size() - 1);
}
};