剑指offer_二叉树的下一个结点

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

知识点
二叉树

思路
解法一、按照中序遍历的顺序将每一个结点存入ArrayList,再遍历list找到当前结点的下一个结点,返回;代码略。

解法二、穷举所有条件;
主要分为1.有右子树 和2.无右子树两种情况:

  1. 有右子树时,找到右子树中的最左边的叶子结点返回;
  2. 没有右子树时,若没有父结点,则直接返回null;
  3. 没有右子树时,若有父结点,则一直向上走,直到此结点node是父结点nextNode的左子结点,返回node;
  4. 没有右子树时,若有父结点但node一直不是nextNode的左子结点(例如pNode是一棵树的最右叶子结点,是中序遍历的最后一个结点),则返回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 if(pNode.next!=null){
            TreeLinkNode nextNode = pNode.next;
            TreeLinkNode node = pNode;
            while(node!=nextNode.left){
                node = node.next;
                nextNode = nextNode.next;
                if(nextNode==null) return null;
            }
            return nextNode;
        }
        else return null;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值