题目:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:
下图中序遍历为:DBFEGAC
- 第一种情况:如果给定节点的右子树不为空,那么该节点的下一个节点是右子树的最左节点;
例如:给定B,则下一个节点为F - 否则: 第二种情况:没有右子树时,那么就得找父节点,当父节点的左等于当前节点也就找到了
1.当前节点可能为上图的D,则下一个节点为B,return B
2.当前节点可能为上图的G,则下一节点为A,return A
3.当前节点可能为上图的C,则下一节点为nul,因为已经遍历完了 上图中序遍历为:DBFEGAC
js代码
function GetNext(pNode) {
if (!pNode) {
return null;
}
//如果当前结点右子树不空,那么中序下一个节点就是右子树的最左节点;如果右子树没有左子树就直接返回右子树根节点
if (pNode.right!=null) {
pNode=pNode.right;
while(pNode.left) { //一直找到最下左子树
pNode=pNode.left;
}
return pNode;
}
//右节点为空的两种情况
//1、
//如果 父节点的右节点 等于 当前的节点,继续返回到上层的父节点...直到 父节点左子节点 等于 当前节点
while (pNode.next!=null&&pNode.next.right==pNode) {
pNode=pNode.next;
}
//2、
//如果右子树为空,返回上层父节点
return pNode.next;
}
}