题目描述:(来源:牛客网)
标签:二叉树
给定一个二叉树其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的next指针。
代码:
思路分析:
1、假如当前结点有右子树,则它的下一个结点是右子树最左边的结点
2、假如当前结点没有右子树,则它的下一个结点可能是它的父节点
①如果父节点.left=当前结点,则返回父节点
②如果不等,说明是它的右儿子得到的父节点,则一直回溯到根节点,返回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.right != null){
TreeLinkNode node = pNode.right;
while(node.left != null){
node = node.left;
}
return node;
}else{
while(pNode.next != null){
TreeLinkNode parent = pNode.next;
if(parent.left == pNode){
return parent;
}
pNode = pNode.next;
}
}
return null;
}
}