题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路
分情况:
- 当前节点有右子树>>>>>右子树的最左那个节点
- 当前节点没有右子树:
2.1当前节点是左节点>>>>>父节点
2.2当前节点是右节点>>>>>父辈节点中是左节点的父节点
/*
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 pNext = null;
//1.如果当前节点有右子树
if(pNode.right!=null){
pNext = pNode.right;
while(pNext.left!=null){
pNext = pNext.left;
}
return pNext;
}//2.如果没有右子树
else if(pNode.next !=null){
//2.1当前节点是左子树
if(pNode.next.left == pNode){
pNext = pNode.next;
}else{
//2.2当前节点是右子树
TreeLinkNode cur = pNode;
TreeLinkNode par = pNode.next;
while(par!=null&&cur == par.right){
cur = par;
par = par.next;
}
pNext = par;
}
return pNext;
}
return null;
}
}