题目:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
解析:给出二叉树的当前结点,获取中序遍历顺序中当前结点的下一结点
①首先判断当前结点是否有当前结点;
②如果当前结点有右结点,则以该右结点为根节点,遍历根节点直到最左节点;
③如果当前结点没有右结点:(有两种情况)
a.如果当前结点是根节点,那么中序遍历后的下一结点为null;
b.沿当前结点往父节点向上遍历,直到找到一个父节点的左子节点为该孩子结点(为什么是这样?把这个规律带到中序遍历中我们发现当前结点的下一结点的确是这样得到的)
//pNode 是当前的结点 GetNext返回中序遍历中pNode的下个结点
public TreeLinkNode GetNext(TreeLinkNode pNode){
if (pNode == null) return null;
//1如果当前结点有右节点
if (pNode.right != null){
pNode = pNode.right;
//这是个小坑,pNode必须不为null
while (pNode.left!=null){
pNode = pNode.left;
}
return pNode;
}else{
//2如果当前结点没有右结点
TreeLinkNode parent = pNode.next;//当前结点的父节点
//2.1如果当前结点为根节点,下一个节点为null
if (parent == null) return null;
//2.2沿当前节点向上遍历找到一个节点的父节点的左子树为该节点,找不到return null
while(parent!= null){
if (parent.left==pNode) return parent;
pNode = parent;
parent = parent.next;
}
return null;
}
}