【题目】
现在有一种新的二叉树节点类型如下:
public class Node
{
public int value;
public Node left;//左节点
public Node right;//右节点
public Node parent;//父节点
public Node(int data)
{ this.value = data; }
}
头节点的parent指向null。
只给一个在二叉树中的某个节点 node,请实现返回node的后继节点的函数。
在二叉树的中序遍历的序列中, node的下一个节点叫作node的后继节点,node的前一个结点叫做node的先驱结点。
【分析】
后继结点
1.如果这个结点有右子树,那么说明这个节点充当的就是中间位置,他的后继结点一定是右子树上最左边的节点。
2.如果这个节点没有右子树,我们可以认为他是左边的最后一个结点,那么中间位置的就是他的后继结点。就相当于我们再找这个节点是作为那个节点的左子树的最后一个节点(左子树上根节点的右节点是最后一个)
先驱结点
1.如果这个节点有左子树,那么说明这个节点充当的就是中间位置,他的先驱结点是左子树上最后一个位置(左子树上根节点的右节点)。
2.如果这个节点没有左子树,说明他是充当的是右子树上的第一个位置,他的先驱结点是哪一个结点把它作为右子树上第一个位置。
【代码实现过程】
【后继结点】
public static Node getSuccessorNode(Node node) {
if(node==null) {
return null;
}
//1.如果有右子树,后继结点是右子树上最左边的节点
//2.如果没有右子树,后继结点是把他作为左子树上最右边节点的那个节点(第一步是找到)
if(node.right!=null) {
return getLeftMost(node.right);
}else {
Node parent=node.parent;//得到第一个父节点
while(parent.left!=node&&parent!=null) {
node=parent;//当前节点走到父节点的位置
parent=node.parent;//父节点再往上走一级
}
return parent;
}
}
//3.获得右子树上最左边的节点(最左边的节点)
public static Node getLeftMost(Node node) {
if (node==null) {
return null;
}
else {
while(node.left!=null) {
node=node.left;
}
return node;
}
}