[剑指Offer] 二叉树的下一个结点(Python)

题目描述

给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。

思路

分情况依次判断。
1. 若给定结点为空节点，则返回空
2. 若给定结点有右子树，则将结点指向右子树的根节点，然后找到其最左边的叶节点并返回
3. 若给定结点没有父节点，则返回空
4. 若给定结点有父节点，若给定结点是其父节点的左孩子，则返回其父节点；否则循环地找父节点的父节点直到当前结点是其父节点的左孩子，则返回其父节点。

代码

# -*- coding:utf-8 -*-
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None
class Solution:
    def GetNext(self, pNode):
        # write code here
        if pNode is None:
            return None

        while pNode.right is not None:
            pNode = pNode.right
            while pNode.left is not None:
                pNode = pNode.left
            return pNode

        while pNode.next is not None:
            if pNode.next.left is pNode:
                return pNode.next

        return None