题目:
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
解题思路:
首先把中序遍历的结果显示出来。—根 左 右
中序遍历后的结果为:DNH B IEJ A FK C LGM
从这个结果来分析中序遍历后的下一个节点情况:
1、二叉树为空,返回空;
2.有右子树的情况:
其下一个节点就是其右子树的最左边的节点,一路向下寻找当前节点的右子树的最左边的节点。
3.没有右子树的情况:分为两种情况
情况一:
当前节点是其父节点的左孩子,则下一个节点就是它的父节点;
情况二:
当前节点是其父节点的右孩子,则找它的下一个节点就要一直找它当前节点的父节点的父节点的父节点,直到当前节点是其父节点的左孩子(可能会有点绕,可以自己画一个有好几个节点只有右孩子的情况,观察一下是怎样遍历到它的下一个节点的),如果没有找到,则说明当前节点是尾节点。
(情况二的图:G的下一个节点是C)
代码:
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;
}
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;//到了根节点,都还没有找到,则返回null
}
}