给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
一刷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 pNode;
TreeLinkNode node = pNode;
if(node.right != null){
node = node.right;
while(node.left != null) node = node.left;
return node;
}else{
while(node.next != null){
if(node.next.left == node) return node.next;
node = node.next;
}
}
return null;
}
}