要求
给定一棵二叉搜索树和其中的一个节点 p ，找到该节点在树中的中序后继。如果节点没有中序后继，请返回 null 。
节点 p 的后继是值比 p.val 大的节点中键值最小的节点。
示例 1：
输入：root = [2,1,3], p = 1
输出：2
解释：这里 1 的中序后继是 2。请注意 p 和返回值都应是 TreeNode 类型。
示例 2：
输入：root = [5,3,6,2,4,null,null,1], p = 6
输出：null
解释：因为给出的节点没有中序后继，所以答案就返回 null 了。
提示：
- 树中节点的数目在范围 [1, 104] 内。
- -105 <= Node.val <= 105
- 树中各节点的值均保证唯一。
核心代码
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
def inorderSuccessor(self, root: 'TreeNode', p: 'TreeNode') -> 'TreeNode':
if not root:
return None
mapping = {}
def inorder(node):
if not node:
return []
mapping[node.val] = node
return inorder(node.left) + [node.val] + inorder(node.right)
Inorder = inorder(root)
for i,val in enumerate(Inorder):
if val == p.val:
return mapping[Inorder[i + 1]] if i != len(Inorder) - 1 else None
另一解法
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
def inorderSuccessor(self, root: 'TreeNode', p: 'TreeNode') -> 'TreeNode':
if not root:
return None
def inorder(node):
if not node:
return []
return inorder(node.left) + [node] + inorder(node.right)
Inorder = inorder(root)
for i,node in enumerate(Inorder):
if node.val == p.val:
return Inorder[i + 1] if i != len(Inorder) - 1 else None
解题思路：第一种解法：我们对树进行中序遍历，然后我们使用一个字典来存储值和节点信息，在后面搜索的时候使用；第二种解法：是对第一种解法的简化，我们直接存储节点信息即可我们不用做值：节点的对应关系.
