写在前面:
这道问题,由于是中序遍历,我们需要考虑以下几种情况:
(1)root=null,那么直接返回null
(2)如果node有右子树,那么就从node的右子树开始寻找,一直找到node的右子树的最左结点,即为node中序遍历的下一个结点。
(3)如果node没有右子树,那么就找node的父结点的left,是否等于node,如果等于node,就返回这个父亲结点,不等于就为第四种情况
(4)如果node没有右子树,且node的父结点的left不等于node,那么就找node父结点的父结点
代码实现如下:
public TreeLinkNode GetNext(TreeLinkNode node) {
// 注意:结点的next指向父亲结点
// (1)第1种情况
if (node == null) {
return null;
}
// (2)第2种情况
if (node.right != null) {
node = node.right;
while (node.left != null) {
node = node.left;
}
return node;
}
// (3)第3种情况
while (node.next != null) {
if (node.next.left == node) {
return node.next;
} else {
// (4)第4种情况
node = node.next;
}
}
return null;
}