剑指Offer_二叉树的下一个结点

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

参考思路
       由于这个二叉树提供了结点指向父结点的指针，因此可以根据给出的结点找到根结点，然后中序遍历将结点顺序保存到List中；最后遍历List找到目标结点，如果不是最后一个则返回其下一个结点，否则返回null。

参考代码

/*
public class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}
*/
import java.util.*;
public class Solution {
    ArrayList<TreeLinkNode> arr = new ArrayList<TreeLinkNode>();
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        TreeLinkNode root = pNode;
        while(root.next != null){
           root = root.next;
        }
        midSort(root);
        for(int i=0;i<arr.size();i++){
            if(arr.get(i) == pNode){
                return i == arr.size()-1 ? null : arr.get(i+1);
            }
        }
        return null;
    }
    
    public void midSort(TreeLinkNode pNode){
        if(pNode != null){
            midSort(pNode.left);
            arr.add(pNode);
            midSort(pNode.right);
        }
    }
}