剑指offer 58题 【树】二叉树的下一个结点

题目描述

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

牛客传送门：点击打开链接

思路：

①判断当前节点是否有右子树，如果有，那么右子树最左节点就是返回值。

②如果没有右子树，则判断父节点的左子树是不是当前节点，若是则返回父节点；若不是则把当前节点设置为父节点，继续这个判断。

class TreeLinkNode {    int val;    TreeLinkNode left = null;    TreeLinkNode right = null;    TreeLinkNode next = null;    TreeLinkNode(int val) {        this.val = val;    }}public class Title58 {    public TreeLinkNode GetNext(TreeLinkNode pNode)    {        if(pNode == null)            return null;        TreeLinkNode ret = null;        // 如果当前节点的右子树不为空，  则下一个节点是右子树的最左子节点        if(pNode.right != null){            ret = pNode.right;            while(ret.left != null)                ret = ret.left;                        return ret;        }                if(pNode.next != null){            ret = pNode.next;            while(ret != null && ret.left != pNode){ // 这里注意ret为空的情况                pNode = ret;                ret = ret.next;            }                        return ret;        }        return ret;    }        public static void main(String[] args) {        TreeLinkNode a = new TreeLinkNode(6);        TreeLinkNode b = new TreeLinkNode(2);        TreeLinkNode c = new TreeLinkNode(8);        TreeLinkNode d = new TreeLinkNode(1);        TreeLinkNode e = new TreeLinkNode(4);        TreeLinkNode f = new TreeLinkNode(7);        TreeLinkNode g = new TreeLinkNode(9);        TreeLinkNode h = new TreeLinkNode(3);        TreeLinkNode i = new TreeLinkNode(5);        TreeLinkNode j = new TreeLinkNode(10);        a.left = b;a.right = c;        b.left = d;b.right = e;        c.left = f;c.right = g;        e.left = h;e.right = i;        g.right = j;                b.next = a;c.next = a;        d.next = b;e.next = b;        f.next = c;g.next = c;        h.next = e;i.next = e;        j.next = g;                // 进行一次中序遍历        List<TreeLinkNode> list = new ArrayList<TreeLinkNode>();        new Title58().print(a,list);        // 输出值：1 2 3 4 5 6 7 8 9 10        list.forEach( n -> System.out.print(n.val + " ") );        // 另一种lambda写法        // List<Integer> numList = list.stream().map((r) -> r.val).collect(Collectors.toList());        // System.out.println(numList);                // 遍历list，输出当前节点的下一节点val        // 输出值： 2 3 4 5 6 7 8 9 10 null        for(TreeLinkNode node : list){            TreeLinkNode next = new Title58().GetNext(node);            if(next != null)                System.out.print(next.val + " ");            else                System.out.println("null");        }    }        public void print(TreeLinkNode root,List<TreeLinkNode> list){        if(root.left != null)            print(root.left,list);        list.add(root);        if(root.right != null)            print(root.right,list);    }}