二叉树的下一个结点

思路一：

因为每个结点都包含指向父节点的指针，所以可以先找到父节点，然后再进行中序遍历，找到给点结点的下一个结点。

import java.util.ArrayList;

/*
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)
    {
        //1.先找到根结点
        TreeLinkNode root = findRoot(pNode);
        //2.中序遍历找到下一个结点
        ArrayList<TreeLinkNode> linkNodes = new ArrayList<TreeLinkNode>();
        inOrder(root,linkNodes);
        int index = 0;
        for (int i = 0; i < linkNodes.size(); i++) {
            if (linkNodes.get(i).val == pNode.val){
                index = i + 1;
                break;
            }
        }
        return index >= linkNodes.size() ? null : linkNodes.get(index);
    }

    private void inOrder(TreeLinkNode root, ArrayList<TreeLinkNode> linkNodes) {
        if (root != null){
            inOrder(root.left,linkNodes);
            linkNodes.add(root);
            inOrder(root.right,linkNodes);
        }
    }


    private TreeLinkNode findRoot(TreeLinkNode pNode) {
        if (pNode.next == null){
            return pNode;
        }
        return findRoot(pNode.next);
    }
}

思路二：

若该节点存在右子树：则下一个节点为右子树最左子节点。

若该节点不存在右子树：这时分两种情况：

• 该节点为父节点的左子节点，则下一个节点为其父节点
• 该节点为父节点的右子节点，则沿着父节点向上遍历，直到找到一个节点的父节点的左子节点为该节点，则该节点的父节点为下一个节点

import java.util.ArrayList;

/*
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;
        }
        //1.若该节点存在右子树：则下一个节点为右子树最左子节点
        if (pNode.right != null){
            pNode = pNode.right;
            while (pNode.left != null){
                pNode = pNode.left;
            }
            return pNode;
        }else if (pNode.next != null && pNode == pNode.next.left){
            //2. 该节点为父节点的左子节点，则下一个节点为其父节点
            return pNode.next;
        }else if (pNode.next != null && pNode == pNode.next.right){
            //3.该节点为父节点的右子节点，则沿着父节点向上遍历，知道找到一个节点的父节点的左子节点为该节点，则该节点的父节点下一个节点
            while (pNode.next != null && pNode.next.left != pNode){
                pNode = pNode.next;
            }
            return pNode.next;
        }else {
            return pNode.next;
        }
    }
}