「这是我参与11月更文挑战的第24天,活动详情查看:2021最后一次更文挑战」
题目:将有序数组转为二叉搜索树
给你一个整数数组 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,3] 和 [3,1] 都是高度平衡二叉搜索树。
提示:
- 1 <= nums.length <= 104
- -104 <= nums[i] <= 104
- nums 按 严格递增 顺序排列
解题思路
首先要理解什么是二叉搜索树
二叉搜索树:左子树节点都小于根节点值,右子树节点都大于根节点值。
由于nums已经升序排列,要构造高度平衡二叉树,所以只需将中间值作为根节点,左边数组的数去构造左子树,右边数组的数去构造右子树,递归构造。
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {number[]} nums
* @return {TreeNode}
*/
const sortedArrayToBST = nums => {
if (!nums.length) return null;
// 去中间索引
const mid = nums.length >> 1;
// 构建二叉树,根节点为中间值,左子树为左侧值继续构建,右子树为右子树继续构建
const root = new TreeNode(
nums[mid],
sortedArrayToBST(nums.slice(0, mid)),
sortedArrayToBST(nums.slice(mid + 1))
);
return root;
};
结束语
这里是小葵🌻,只要把心朝着太阳的地方,就会有温暖~
让我们一起来攻克算法难关吧!!