剑指offer57--二叉树的下一个节点

时间限制：1秒 空间限制：32768K 热度指数：111691

 算法知识视频讲解

题目描述

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

二叉树的中序遍历：左中右。所以这个题要考虑三个点

1.如果该节点有右子树，则下一个节点是左子树最左边的一个节点。

2，如果该节点没有右子树，并且该节点是父亲节点的左孩子，那么下个节点是他的父亲节点。

3.如果该节点没有右子树，并且该节点是父亲节点的右孩子，则找到该节点的父亲节点，如果父亲节点是父亲节点的父亲节点的左孩子，则还回父亲节点的父亲节点，否则，继续3

结合图，我们可发现分成两大类：1、有右子树的，那么下个结点就是右子树最左边的点；（eg：D，B，E，A，C，G） 2、没有右子树的，也可以分成两类，a)是父节点左孩子（eg：N，I，L） ，那么父节点就是下一个节点 ； b)是父节点的右孩子（eg：H，J，K，M）找他的父节点的父节点的父节点...直到当前结点是其父节点的左孩子位置。如果没有eg：M，那么他就是尾节点。

/*
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}
*/
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
            //如果该节点有右子数，则找出右字数最左边的节点 
		
		 if(pNode.right!=null) {
			 TreeLinkNode cur=pNode.right;
			 while(cur.left!=null) {
				 cur=cur.left;
			 }
			 return cur;
		 }
			
	     while(pNode.next!=null) {
	    	 if(pNode.next.left==pNode) return pNode.next;
	    	 pNode=pNode.next;
	     }
		return null;
    }
}