题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。
思路:
递归:
首先通过next指针找到二叉树的根结点,再利用根结点作中序遍历,获得中序遍历的结点列表,最后通过遍历列表找到所给结点的中序遍历下一结点。
需注意的是:边界情况,所给结点可能是中序遍历的尾结点,这种情况其下一结点为null.
/*
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
*/
import java.util.ArrayList;
public class Solution {
public TreeLinkNode GetNext(TreeLinkNode pNode)
{
TreeLinkNode root=pNode;
while(root.next!=null){
root=root.next;
}
ArrayList<TreeLinkNode> list=new ArrayList<>();
list=inOrder(root,list);
for(int i=0;i<list.size();i++){
if(pNode==list.get(i)){
if(i==list.size()-1){//注意点,考虑边界,所给结点可能是最后一个结点
pNode=null;
}else{
pNode=list.get(i+1);
}
break;
}
}
return pNode;
}
//中序遍历
public ArrayList<TreeLinkNode> inOrder(TreeLinkNode root,ArrayList<TreeLinkNode> list){
if(root==null)return list;
inOrder(root.left,list);
list.add(root);
inOrder(root.right,list);
return list;
}
}
非递归:
分析二叉树的下一个节点,一共有以下情况:
1.二叉树为空,则返回空;
2.节点右孩子存在,则设置一个指针从该节点的右孩子出发,一直沿着指向左子结点的指针找到的叶子节点即为下一个节点;
3.节点右孩子不存在,如果该节点是其父节点的左孩子,则返回父节点;否则继续向上遍历其父节点的父节点,重复之前的判断,返回结果。
/*
public class TreeLinkNode {
int val;
TreeLinkNode left = null;
TreeLinkNode right = null;
TreeLinkNode next = null;
TreeLinkNode(int val) {
this.val = val;
}
}
*/
import java.util.ArrayList;
import java.util.Iterator;
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;
}
}else{
while(pNode.next!=null&&pNode.next.left!=pNode){
pNode=pNode.next;
}
pNode=pNode.next;
}
return pNode;
}
}