这个题目就是考察中序遍历,就是分三种情况:
- 当结点右子树不为空时,则下一个一定是右子树的最左边的结点
- 当结点是父节点的左子树时,那下一个一定是父节点
- 当结点时父节点的右子树时,则顺着向上找,直到一个目标结点是它父节点的左子树,那下一个结点就是这个目标结点的父节点。
代码如下:
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
if (pNode.right != null) { //第一种情况
TreeLinkNode result = pNode.right;
while(result.left != null) { //找右子树的最左结点
result = result.left;
}
return result;
}
TreeLinkNode pParent = pNode.next;
if (pParent == null) //如果父节点为null,则直接返回null说明是根节点
return null;
if (pParent.right == pNode) {
//找到一个目标节点,且它是父节点的左子树,返回它的父节点,第三种情况
while(pParent.next != null && pParent != pParent.next.left) {
pParent = pParent.next;
}
return pParent.next;
} else { //否则就是父节点,第二种情况
return pParent;
}
}
}