后继节点定义:
二叉树以中序的方式进行遍历打印,节点X的下一个节点,就是X的后继节点。
假设二叉树如下图所示:则中序遍历的后打印出来的就是 4 -> 2 -> 5 -> 1 -> 6 -> 3 -> 7。如果X = 3,则X的后继节点就是7。
如果Node的结构中只有left和right的话,那时间复杂度就是
O
(
N
)
O(N)
O(N),从给定的头结点开始遍历,先打印出来中序,在遍历中序找到X,在找到X的下一个。
如果给定的Node的数据结构中包含parent指针,则可以将时间复杂度降低到
O
(
K
)
O(K)
O(K)。K为X到后继节点的实际距离。
public static class Node {
public int value;
public Node left;
public Node right;
public Node parent;
public Node(int data) {
this.value = data;
}
}
分析:
因为中序遍历是按照 左 -> 头 -> 右的方式进行打印,所以要考虑X所处的位置。
- 如果X有右树,则X的后继节点一定是右树中的最左节点。
- 如果X没有右树,则看X是否是X.parent的left节点,如果是,则parent就是X的后继节点,如果不是,则一直向上找。直到找到当前节点是parent的左子节点为止,如果找不到,则说明X是右树最末端节点,后继节点为null。
代码实现:
public static Node getSuccessorNode(Node node){
if (node == null){
return node;
}
if (node.right != null){
return getLeftMost(node.right);
}else{
Node parent = node.parent;
if (parent != null && node != parent.left){
node = parent;
parent = node.parent;
}
return parent;
}
}
public static Node getLeftMost(Node node){
if (node == null){
return node;
}
while (node.left != null){
node = node.left;
}
return node;
}