思路:
- 1.如果有右孩子,后继节点就是最左边的
- 2.如果没有右孩子,判断是否是父节点的左孩子,是的话,返回,不是继续往上找
- 3.找不到就是null
即:如果一个结点有右子树,就将它右子树上最左的结点返回,否则将当前结点的指针往上窜,当窜到它是它父结点的左结点时停止,此时那个父结点就是原结点的后继
/*
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){
pNode=pNode.right;
while(pNode.left!=null){
pNode=pNode.left;
}
return pNode;
}
//将当前结点的指针往上窜,当窜到它是它父结点的左结点时停止,此时那个父结点就是原结点的后继
else{
TreeLinkNode parent=pNode.next;
while(parent!=null && parent.left!=pNode){
pNode=parent;
parent=parent.next;
}
return parent;
}
}
}
京公网安备 11010502036488号