算法笔记
二叉树的下一个结点
图是中序遍历图,红色箭头指向中序遍历的下一个节点,从上面的图可以看出来,基本上查找当前节点的中序遍历的下一个节点的规律如下:
- 有右子树的,那么下个结点就是右子树最左边的点;(eg:D,B,E,A,C,G)
- 没有右子树的,也可以分成两类,a)是父节点左孩子(eg:N,I,L) ,那么父节点就是下一个节点 ; b)是父节点的右孩子(eg:H,J,K,M)找他的父节点的父节点的父节点…直到当前结点是其父节点的左孩子位置。如果没有eg:M,那么他就是尾节点。
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode) {
if(pNode == null) return null;
//如果节点的右孩子不为空的话
if(pNode.right != null){
pNode = pNode.right;
while(pNode.left != null){
pNode = pNode.left;
}
return pNode;
}
//如果节点没有右孩子
//找父节点,直到该节点是父节点的左孩子为止
while(pNode.next != null){
if(pNode.next.left == pNode){
return pNode.next;
}
pNode = pNode.next;
}
//如果都没有,那就返回空
return null;
}
}