剑指offer41 — 二叉树的下一个结点
题目
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
分析
中序遍历即为左中右的顺序。将结点分为两种情况,有右子树的情况以及没有右子树的情况。
- 寻找右子树,如果存在就一直找到右子树的左子树,就是输出的下一个结点
- 没有右子树,就寻找他的父节点,一直找到它是父节点的左子树的时候,打印该左子树结点。
代码
# -*- 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.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