剑指Offer:二叉树的下一个节点(8)

题目描述:

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

解题思路:

如图所示:

中序遍历的顺序为:{d,b,h,e,i,a,f,c,g}

中序遍历分为三种情况:

1. 如果当前结点有右子树

下一个节点就是它的右子树的最左子节点(从右子节点出发一致沿着指向左子节点的指针,即可找到)
如:b ——>h, a——>f

2. 如果当前节点没有右子树

(1) 如果当前节点是它父节点的左子节点

下一个节点就是它的父节点
如:d——>b, f——>c, h——>e

(2)如果当前节点是它父节点的右子节点

沿着指向父节点的指针一直向上遍历,直到找到一个是它父节点的左子节点的节点,如果这样的节点存在,那么这个节点的父节点就是我们要找到的下一个节点
如:i节点的下一个节点:沿着指向父节点的指针向上遍历,先到e,e是b的右子节点,不是,继续向上遍历,到b, b是a的左子节点,因此节点b的父节点a就是i的下一个节点

 1 public class Solution {
 2     public TreeLinkNode GetNext(TreeLinkNode pNode)
 3     {
 4         //树为空
 5         if(pNode == null)
 6             return null;
 7         TreeLinkNode pNext = null;//当前节点的下一节点
 8         //如果当前节点有右子节点,下一节点为右子树的最左子节点
 9         if(pNode.right != null){
10             TreeLinkNode pRight = pNode.right;//当前节点的右子节点
11             while(pRight.left != null)
12                 pRight = pRight.left;
13             pNext = pRight;
14         }else if(pNode.next != null){
15             TreeLinkNode pCur = pNode;//当前节点
16             TreeLinkNode pParent = pNode.next;//当前节点的父节点
17             //若当前节点是其父节点的右子节点,遍历,直到当前节点没有父节点,或者当前节点是其父节点的左子节点,则下一节点就是其父节点
18             while(pParent != null && pCur == pParent.right){
19                 pCur = pParent;
20                 pParent = pParent.next;
21             }
22             pNext = pParent;
23             
24         }
25         return pNext;
26     }
27 }

 

posted @ 2020-08-11 15:02  手下留情  阅读(115)  评论(0编辑  收藏  举报