给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
分成两大类:
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;
//这里的next是指向父节点
TreeLinkNode(int val) {
this.val = val;
}
}
*/
public class Solution {
//注意题目定义的next是指向父节点的
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
if(pNode==null){
return null;
}
if(pNode.right!=null){//如果有右子树,则找到右子树最左边的节点
pNode=pNode.right;
while(pNode.left!=null){
pNode=pNode.left;
}
return pNode;
}
while(pNode.next!=null)//假如没有右子树,迭代寻找该节点的父节点,如果父节点是某个节点的左子节点,则返回该节点。(也有可能返回空)
{
if(pNode.next.left==pNode)
return pNode.next;
pNode = pNode.next;
}
return null;
}
}