题目描述:给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:
1、找到根结点
2、进行中序遍历将结果存入list中,找到该结点,输入对应的下一个结点。
边界条件:注意当为空树或只有根结点时,没有下一个结点,输出为null。
public class Solution {
ArrayList<TreeLinkNode> list=new ArrayList<>();
public TreeLinkNode GetNext(TreeLinkNode pNode){
//空树或是只有根节点,则没有下一个节点
if(pNode==null || (pNode.left==null && pNode.right==null && pNode.next==null)){
return null;
}
//找到根节点
TreeLinkNode root=pNode;
while(true){
if(root.next==null){
break;
} else{
root=root.next;
}
}
//找到根节点,进行中序遍历,将中序遍历结果存入list中
infixNode(root);
//找到对应的节点,进行遍历
for(int i=0;i<list.size();i++){
if(list.get(i)==pNode && i!=list.size()-1){
return list.get(i+1);
}
}
return null;
}
//中序遍历
public void infixNode(TreeLinkNode root){
if(root.left!=null){
infixNode(root.left);
}
list.add(root);
if(root.right!=null){
infixNode(root.right);
}
}
}