08-二叉树的下一个节点

2020-05-05  本文已影响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;
    }
}
上一篇下一篇

猜你喜欢

热点阅读