思路:根据当前结点pNode的位置,分为以下几种情况:
(1)pNode有右子树:以pNode的右孩子为起点,寻找最左端的结点,返回。
(2)pNode没有右子树:
1)pNode没有父节点,返回空;
2)pNode有父节点,并且pNode是父节点的左孩子,返回父节点;
3)pNode有父节点,并且pNode是父节点的右孩子,此时说明pNode与pNode的父节点都已经访问过了:
寻找到不是父节点右孩子的父节点(有点绕),即沿着右子树向左找;返回最终的父节点。
public static TreeLinkNode GetNext(TreeLinkNode pNode)
{
if(pNode==null) return null;
//有右子树
if(pNode.right!=null){
pNode=pNode.right;
while(pNode.left!=null){
pNode=pNode.left;
}
return pNode;
}
//没有右子树,并且没有父节点
if(pNode.next==null) return null;
//没有右子树,有父节点,并且是父节点的左孩子
if(pNode.next.left==pNode) return pNode.next;
//没有右子树,有父节点,并且是父节点的右孩子
if(pNode.next.right==pNode){
pNode=pNode.next;
while(pNode.next!=null&&pNode.next.right==pNode){
pNode=pNode.next;
}
return pNode.next;
}
return null;
}