题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
问题分析
二叉树的结构如下所示,题目中的描述next意思是指向当前节点的父节点。我们要知道中序遍历的含义,先按中序遍历左子树,再输出当前结点,然后再中序遍历右子树。
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
代码及讲解
按照中序遍历的特点,如果当前结点没有右子树,且它是它父结点的左子结点(例如D),则当前结点的下一个遍历的结点就是它的父结点(B),如果它是它父结点的右子结点(G,I)则它应该向上寻找到一个结点是该结点父结点的左子结点(G对应null,I对应A),则下一个遍历的就是该结点的父节点。
如果当前结点有右子树(如E,B),则按照中序遍历的规则,应该对当前节点的右子结点进行中序遍历,则下一个遍历的点为当前节点右子树中最下面的左子节点(H)。
我这么说肯定看的很茫然,找了个图当例子讲一下。
下图的中序遍历顺序为{D,B,H,E,I,A,F,C,G}。配合上面的讲述应该比较清楚了吧。
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;
}
while(pNode.next!=null){
if(pNode.next.left==pNode)return pNode.next;
if(pNode.next.right==pNode) pNode=pNode.next;
}
return null;
}
}
总结
我觉得本题要注意对问题的分析,我开始写是从是否是叶子结点来判断的,最后写的一塌糊涂,把自己也绕了进去,后来看了一些讨论和题解才解出来。其实二叉树中序遍历找下一个结点并不难,注意找到问题的切入点才是关键,老板们加油,我有个朋友说过种一棵树最好的时间是十年前然后就是。