题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
解题思路
- 这个题其实求的是中序线索二叉树某一结点的下一个结点
- 如果一个结点有右子树,那么中序遍历中该结点的下一个结点为右子树中最左边的结点
- 如果一个结点没有右子树,且它的父节点的左子树是该结点,那么父节点是该结点在中序遍历中的下一个结点
- 如果一个结点没有右子树,且它的父节点的左子树不是该结点,则一直向上寻找父节点,直到找到一个是它父节点的左子节点的结点,该结点即为所求。
- 其他情况为 null。
AC代码
/*
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
if(pNode == null) return null;
TreeLinkNode tmp = null;
//有右子树
if(pNode.right != null){
tmp = pNode.right;
while(tmp.left != null) tmp = tmp.left;
return tmp;
}
//有父节点
if(pNode.next != null){
//父节点的左节点是该节点
tmp = pNode.next;
if(tmp.left == pNode) return tmp;
while(tmp != null && tmp.left != pNode){
pNode = tmp;
tmp = tmp.next;
}
return tmp;
}
return null;
}
}