Skip to content
本页目录

0108-将有序数组转换为二叉搜索树

https://leetcode.cn/problems/convert-sorted-array-to-binary-search-tree

给你一个整数数组 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 <= 10^4
- -10^4 <= nums[i] <= 10^4
- nums 按 严格递增 顺序排列

思路

有序数组从中间取值,作为根节点,然后左右分别构造子树

参考代码

csharp
public class Solution {
    public TreeNode SortedArrayToBST(int[] nums) {
    	TreeNode node = null;
    	if(nums.Length == 1){
    		node = new TreeNode(nums[0]);
    		return node;
    	}
    	else if(nums.Length == 2){
    		node = new TreeNode(nums[0]);
    		node.right = new TreeNode(nums[1]);
    		return node;
    	}
    	else if(nums.Length > 2){
    		//取出中点
    		int mid = nums.Length / 2;
    		node = new TreeNode(nums[mid]);
    		//构造左数组
    		int[] leftNums = new int[mid];
    		for(int i=0; i<mid; i++){
    			leftNums[i] = nums[i];
    		}
    		node.left = SortedArrayToBST(leftNums);
    		//构造右数组
    		int[] rightNums = new int[nums.Length - mid - 1];
    		for(int i=0; i<rightNums.Length; i++){
    			rightNums[i] = nums[i+mid+1];
    		}
    		node.right = SortedArrayToBST(rightNums);
    	}
    	return node;
    }
}

Released under the MIT License.