一 题目
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
二 例子
对于下面的二叉树,8的后继节点为9(题目说的中序遍历),6的后继节点为7,5的后继节点为6
三 思路
思路很简单,
①如果节点有右子树,则该节点的后继节点就是往右子树出发,然后转到右子树的左子树,一直到左子树的左子树为空(即输入节点的右子树的最左子树,例如节点8,后继节点就是9)
②如果节点没有右子树,则向上寻找父节点,直到父节点的左子树等于当前节点,则该父节点就是后继节点(如节点7,没有右子树,则向上找,这时到6这个节点,因为6的左子树不等于7,所以继续向上找,这时到节点8,8的左子树等于当前节点6,所以返回8)
四 程序源码
-
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== null) return null;
-
if (pNode.right != null) {
-
pNode = pNode.right;
-
while(pNode!= null&&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;
-
}
-
}
-