题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
题解:
①如果一个节点有右子树,那么它的下一个节点就是它的右子树中的最左子节点。
②一个节点没有右子树:
①如果节点是父节点的左子节点,它的下一个节点是他的父节点
②如果节点是父节点的右子节点,沿着指向父节点的指针一直往上遍历,直到找到一个是它父节点的左子节点的节点
因为是中序遍历所以跟有无左子树无关系,只需要讨论有无右子树的情况。
①有右子树:向下找
②无右子树:向上找
/*
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 pNode;
if(pNode.right!=null){//有右子树
TreeLinkNode cur = pNode.right;
while(cur.left!=null){
cur = cur.left;
}
return cur;
}
else if(pNode.next!=null){//无右子树
TreeLinkNode cur = pNode;
TreeLinkNode tmp = pNode.next;//父节点
if(tmp.left==pNode){
return tmp;
}
while(tmp!=null&&cur==tmp.right){
cur = tmp;
tmp = tmp.next;
}
return tmp;
}
return null;
}
}