牛客66道编程题——二叉树的下一个结点

题目

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

方法一

思路：
若该结点有右子树，寻找右子树。如果右子树一直存在，就一直寻找到右子树的最左边便是下一个结点。

若无右子树，便寻找他的父结点，一直找到它是父结点的左子树，打印父结点。

class Solution:
    def GetNext(self, pNode):
        # write code here
        if pNode.right:
            tmpNode=pNode.right
            while tmpNode.left:
                tmpNode=tmpNode.left
            return tmpNode
        else:
            tmpNode=pNode
            while tmpNode.next:
                if tmpNode.next.left==tmpNode:
                    return tmpNode.next
                tmpNode=tmpNode.next
            return None

方法二

思路：
先写一个中序遍历的算法。关键是从根节点开始遍历，所以第一步还是找到某个节点的根节点，方法是一直使用next判断即可。
再将从根节点中序遍历的结果保存到一个数组中，直接找pNode所在索引的下一个即可。当然要考虑这个节点是不是最后一个，如果是最后一个，直接返回None。

# -*- coding:utf-8 -*-
# class TreeLinkNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#         self.next = None
class Solution:
    
    def __init__(self):
            self.res=[]
    def midOrder(self,pRoot):
            if pRoot.left:
                self.midOrder(pRoot.left)
            self.res.append(pRoot)
            if pRoot.right:
                self.midOrder(pRoot.right)
    def GetNext(self, pNode):
        # write code here
        tmp=pNode
        while tmp.next:#找到根结点
            tmp=tmp.next
        self.midOrder(tmp)#获取中序遍历序列
        if self.res.index(pNode)!=len(self.res)-1:
            return self.res[self.res.index(pNode)+1]
        else:
            return None