您的位置:首页 > 其它

【LEETCODE】108-Convert Sorted Array to Binary Search Tree

2015-11-04 20:17 337 查看
Given an array where elements are sorted in ascending order, convert it to a height balanced BST.

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
def sortedArrayToBST(self, nums):
"""
:type nums: List[int]
:rtype: TreeNode
"""

if len(nums)==0:
return None
elif len(nums)==1:
return TreeNode(nums[0])
else:
mid=nums[len(nums)/2]             #找到nums的中位值,作为root
root=TreeNode(mid)
root.left=self.sortedArrayToBST(nums[0:len(nums)/2])                    #比中位值小的放在left子树
root.right=self.sortedArrayToBST(nums[len(nums)/2+1:len(nums)])         #比中位值大的放在right子树
return root
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: