二叉树的下一个结点

题目描述
给定一个二叉树和其中的一个结点,请找出中序遍历顺序的下一个结点并且返回。注意,树中的结点不仅包含左右子结点,同时包含指向父结点的指针。

思路:
递归:
首先通过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;
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值