美文网首页
08-二叉树的下一个节点

08-二叉树的下一个节点

作者: 一方乌鸦 | 来源:发表于2020-05-05 21:59 被阅读0次

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

/*
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) {
        TreeLinkNode res = null;
        if (pNode.right != null) {
            res = pNode.right;
            while (res.left != null) res = res.left;
        } else if (pNode.next != null) {
            TreeLinkNode c = pNode;
            TreeLinkNode p = pNode.next;
            while (p != null && c == p.right) {
                c = p;
                p = p.next;
            }
            res = p;
        }
        return res;
    }
}

相关文章

网友评论

      本文标题:08-二叉树的下一个节点

      本文链接:https://www.haomeiwen.com/subject/leplghtx.html