首页 > 其他 > 详细

递归与二叉树_leetcode450

时间:2019-03-17 14:24:43      阅读:162      评论:0      收藏:0      [点我收藏+]
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None

class Solution1(object):
def deleteNode(self, root, key):
"""
:type root: TreeNode
:type key: int
:rtype: TreeNode
"""
if not root:
return root

if root.val > key:
root.left = self.deleteNode(root.left,key)
return root
elif root.val < key :
root.right = self.deleteNode(root.right,key)
return root

else:
if not root.left:
successor = root.right
root.right = None
return successor
if not root.right:
successor = root.left
root.left = None
return successor

successor = self.minNode(root.right)

successor.right = self.removeMin(root.right)
successor.left = root.left

root.left = None
root.right = None

return successor


def minNode(self,root):
if not root.left:
return root
return self.minNode(root.left)

def maxNode(self,root):
if not root.right:
return root
return self.maxNode(root.right)

def removeMin(self,root):

if not root.left:
successor = root.right
root.right = None
return successor
root.left = self.removeMin(root.left)
return root




def removeMax(self,root):
if not root.right :
successor = root.left
root.left = None
return successor
root.right = self.removeMax(root.right)
return root


class Solution2(object):
def deleteNode(self, root, key):
"""
:type root: TreeNode
:type key: int
:rtype: TreeNode
"""
if not root:
return root

if root.val > key:
root.left = self.deleteNode(root.left,key)
return root
elif root.val < key :
root.right = self.deleteNode(root.right,key)
return root

else:
if not root.left:
successor = root.right
root.right = None
return successor
if not root.right:
successor = root.left
root.left = None
return successor

successor = self.maxNode(root.left)

successor.left = self.removeMax(root.left)
successor.right = root.right

root.left = None
root.right = None

return successor


def minNode(self,root):
if not root.left:
return root
return self.minNode(root.left)

def maxNode(self,root):
if not root.right:
return root
return self.maxNode(root.right)

def removeMin(self,root):

if not root.left:
successor = root.right
root.right = None
return successor
root.left = self.removeMin(root.left)
return root





def removeMax(self,root):
if not root.right :
successor = root.left
root.left = None
return successor
root.right = self.removeMax(root.right)
return root

递归与二叉树_leetcode450

原文:https://www.cnblogs.com/lux-ace/p/10546821.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!