力扣日记1.14-【二叉树篇】108. 将有序数组转换为二叉搜索树

力扣日记：【二叉树篇】108. 将有序数组转换为二叉搜索树

日期：2023.1.14
参考：代码随想录、力扣

108. 将有序数组转换为二叉搜索树

题目描述

难度：简单

给你一个整数数组 nums ，其中元素已经按升序排列，请你将其转换为一棵高度平衡二叉搜索树。

高度平衡二叉树是一棵满足「每个节点的左右两个子树的高度差的绝对值不超过 1 」的二叉树。

示例 1：
在这里插入图片描述

输入：nums = [-10,-3,0,5,9]
输出：[0,-3,9,-10,null,5]
解释：[0,-10,5,null,-3,null,9] 也将被视为正确答案：

示例 2：
在这里插入图片描述

输入：nums = [1,3]
输出：[3,1]
解释：[1,null,3] 和 [3,1] 都是高度平衡二叉搜索树。

提示：

1 <= nums.length <= 10^4
-10^4 <= nums[i] <= 10^4
nums 按严格递增顺序排列

题解

/*** Definition for a binary tree node.* 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 {
#define SOLUTION 2
public:
#if SOLUTION == 1TreeNode* sortedArrayToBST(vector<int>& nums) {if (nums.size() == 0)   return nullptr;// 中节点的位置int index = nums.size() / 2;TreeNode* node = new TreeNode(nums[index]);// 中节点左边和右边的数组分别作为左右子树去构建(左子树为BST + 右子树为BST -> root为BST)// 而且最后一定也是高度平衡二叉树vector<int> leftNum(nums.begin(), nums.begin() + index);    // 左子树：[startIdx, endIdx)vector<int> rightNum(nums.begin() + index + 1, nums.end()); // 右子树node->left = sortedArrayToBST(leftNum); // 左子树(自身也是BST)作为左节点node->right = sortedArrayToBST(rightNum); // 右子树(自身也是BST)作为右节点return node;}
#elif SOLUTION == 2 // 下标索引TreeNode* sortedArrayToBST(vector<int>& nums) {     return buildTree(nums, 0, nums.size() - 1); // 左闭右闭}// 循环不变量，左闭右闭 [left, right]// 返回值为构造的二叉树的根节点，输入为原始数组以及子数组的起始位置TreeNode* buildTree(vector<int>& nums, int left, int right) {// 终止条件if (left - right > 0)   return nullptr; // 相等也是符合条件的,则index = left// 获得子数组的中点位置// int index = (left + right) / 2; // 可能溢出int index = left + (right - left) / 2;TreeNode* node = new TreeNode(nums[index]);// 递归左区间node->left = buildTree(nums, left, index - 1);  // 左闭右闭// 递归右区间node->right = buildTree(nums, index + 1, right);    // 左闭右闭return node;}
#endif
};