swift 算法 简单22. 将有序数组转换为二叉搜索树

知识共享许可协议 Creative Commons

将一个按照升序排列的有序数组,转换为一棵高度平衡二叉搜索树。

本题中,一个高度平衡二叉树是指一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1。

示例:

给定有序数组: [-10,-3,0,5,9],

一个可能的答案是:[0,-3,9,-10,null,5],它可以表示下面这个高度平衡二叉搜索树:

      0
     / \
   -3   9
   /   /
 -10  5

解法:

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     public var val: Int
 *     public var left: TreeNode?
 *     public var right: TreeNode?
 *     public init(_ val: Int) {
 *         self.val = val
 *         self.left = nil
 *         self.right = nil
 *     }
 * }
 */
class Solution {
    func sortedArrayToBST(_ nums: [Int]) -> TreeNode? {
     if nums.isEmpty {
            return nil
        }
        
        return gBST(nums, 0, nums.count);

    }
    func  gBST(_ nums: [Int], _ l: Int, _ r: Int) -> TreeNode? {
        if (l >= r) {
            return nil;
        }
        let mid = l + (r - l) / 2; 
        
        let node =  TreeNode(nums[mid]);
        node.left = gBST(nums, l, mid);
        node.right = gBST(nums, mid + 1, r);
        
        return node;
    }
}

猜你喜欢

转载自blog.csdn.net/huanglinxiao/article/details/91946799