二叉树的下一个结点

剑指offer之——二叉树的下一个结点

思路1：

①：通过给定的结点找出该树的根结点。
②：中序遍历该树并存储在数组中。
③：最后依次读取数组中的结点，找出二叉树的下一个结点。

/*
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 {
    //集合用于存储结点
    ArrayList<TreeLinkNode> list = new ArrayList<>();
    public TreeLinkNode GetNext(TreeLinkNode pNode) {
        //异常处理
        if (pNode == null) {
            return null;
        }
        //找到树的根节点
        TreeLinkNode root = pNode;
        while (root.next != null) {
            root = root.next;
        }
        midOrder(root);
        int len = list.size();
        for (int i = 0; i < len; i ++) {
            if (list.get(i) == pNode) {
                return i == len - 1 ? null : list.get(i + 1);
            }
        }
        return null;
    }
    
    //中序遍历
    public void midOrder(TreeLinkNode root) {
        if (root != null) {
            midOrder(root.left);
            list.add(root);
            midOrder(root.right);
        }
    }
}

时间复杂度：找根结点，做最坏的打算为O(n),中序遍历为O(n)，再次遍历数组最坏的时间复杂度为O（n），故总的时间复杂度为3O(n)。
空间复杂度：存储树中的所有结点，空间复杂度为O(n)。

思路2：

该节点存在右子树：则下一个结点为右子树的最左结点。
该节点右子树不存在：
①该结点为其父节点的左结点，下一个结点为其父结点。
②该节点为其父节点的右结点，寻找下一个结点比较麻烦，应该依次从该结点找其祖先结点，直到有一个结点是其父节点的左节点。

注：黄色为存在右子树的，红色为不存在右子树且结点为父结点的左结点，绿色为不存在右子树且结点为父结点的右结点。

/*
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 == null) {
            return null;
        }
        //存在右子树，下一个结点是右子树的最左结点
        if (pNode.right != null) {
            TreeLinkNode node = pNode.right;
            while (node.left != null) {
                node = node.left;
            }
            return node;
        }
        //不存在右子树
        else {
            //该结点是父节点的左子节点，下一个结点是其父节点
            TreeLinkNode parent = pNode.next;
            if (parent == null || parent.left == pNode) {
                return parent;
            }
            //该节点是父节点的右子结点，下一个结点是，沿着其父节点一直往上找，直到找到一个祖先结点的左结点是其父节点
            else {
            TreeLinkNode pparent = parent.next;
                while (pparent != null && pparent.left != parent) {
                    parent = pparent;
                    pparent = pparent.next;
                }
                 return pparent;
            }
        }
    }
}

时间复杂度：O(n)。
空间复杂度：O(1)。