JZ57二叉树的下一个结点

题解
题目描述：给你一颗二叉树的一个结点，返回中序遍历顺序中这个结点的下一结点。二叉树不仅有左右孩子指针，还有指向父亲结点的指针。

Q1：首先问你一个问题，如果这道题出现在笔试题中，你会用什么方法做？如果出现在面试题中呢？
A1：我想你肯定有点疑惑，同一道题为什么还分出现在笔试题中还是面试题中呢？很显然，笔试题中只要能过就好，设计的算法丑点，慢点也无所畏，不一定需要最优解法，当然前提是能够通过。而面试中就不一样了，显然面试官希望听到最优解法。

方法一：暴力解法(还原二叉树)

1. 分析
   既然给了二叉树的某个结点，且二叉树存储着指向父结点的指针（next），那我们可以先找到根节点，再对树进行中序遍历，最后根据中序遍历结果找到给定结点的下一结点

根据给出的结点求出整棵树的根节点
根据根节点递归求出树的中序遍历，存入vector
在vector中查找当前结点，则当前结点的下一结点即为所求。
虽然有点暴力，但是时间复杂度也是线性的，第一步：最坏为O(N), N为整棵树结点的个数。第二步：O(N), 第三步：最坏为O(N),
所以整的时间复杂度：3*O(N)

import java.util.*;
public class Solution {
    static ArrayList<TreeLinkNode> list = new ArrayList<>();
    public TreeLinkNode GetNext(TreeLinkNode pNode){
       TreeLinkNode parent = pNode;//定义根节点
       //且二叉树存储着指向父结点的指针（next）,那我们可以先找到根节点
       while(parent.next != null){
            parent = parent.next;
        }
        InOrder(parent);//根节点的中序遍历
        for(int i=0; i<list.size(); i++){
        	//找当前结点的下一结
            if(pNode == list.get(i)){
                return i == list.size()-1 ? null : list.get(i+1);
            }
        }
        return null; 
    }
    //递归中序遍历，把每一个节点遍历到list中
     void InOrder(TreeLinkNode pNode){
          if(pNode!=null){
             InOrder(pNode.left);
             list.add(pNode);
             InOrder(pNode.right);
        }
    }
}

时间复杂度还可以接受，关键是思路好想并且每一步的代码都很简单。
时间复杂度：O(N)
空间复杂度：O(N)

方法二：最优解法
但是，如果在面试中，方法一肯定上不了台面。但是最优解法该怎么去想呢？想不出来就画图分析，举个中序遍历的图：如下：

1 => 2 // 显然下一结点是 1 的父亲结点
2 => 3 // 下一节点是当前结点右孩子的左孩子结点，其实你也应该想到了，应该是一直到左孩子为空的那个结点
3 => 4 // 跟 2 的情况相似，当前结点右孩子结点的左孩子为空的那个结点
4 => 5 // 5 是父亲结点 3 的父亲结点，发现和1有点像，因为 1，3,同样是父亲结点的左孩子
5 => 6 // 跟 4=>5 一样的道理
6 => 7 // 跟 3=>4 一样的道理
7 => null // 因为属于最尾结点
此时，可以总结一下：
[1] 是一类：特点：当前结点是父亲结点的左孩子
[2 3 6] 是一类，特点：当前结点右孩子结点，那么下一节点就是：右孩子结点的最左孩子结点,如果右孩子结点没有左孩子就是自己
[4 5]是一类，特点：当前结点为父亲结点的右孩子结点，本质还是[1]那一类
[7]是一类，特点：最尾结点

我写的可能不够清晰，但是，思想你要明白，当遇到不会的题，可以根据题意画图，分析，分析方法是关键。

仔细观察，可以把中序下一结点归为几种类型：

有右子树，下一结点是右子树中的最左结点，例如 B，下一结点是 H

无右子树，且结点是该结点父结点的左子树，则下一结点是该结点的父结点，例如 H，下一结点是 E

无右子树，且结点是该结点父结点的右子树，则我们一直沿着父结点追朔，直到找到某个结点是其父结点的左子树，如果存在这样的结点，那么这个结点的父结点就是我们要找的下一结点。例如 I，下一结点是 A；例如 G，并没有符合情况的结点，所以 G 没有下一结点

2. 代码

import java.util.*;
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode){
      //1.有右子树
      if(pNode.right != null){
          TreeLinkNode pRight = pNode.right;
          while(pRight.left != null){
            pRight = pRight.left;  
          }
          return pRight;
      }
      //没有右子树，返回的是父亲节点
      //父亲节点不为空，且当前节点是父亲节点的左子树
      if(pNode.next != null && pNode.next.left == pNode){
          return pNode.next;
      }
      
      //父亲节点不为空,且结点是该结点父结点的右子树
      //我们一直沿着父结点追朔，直到找到某个结点是其父结点的左子树
      //如果存在这样的结点，那么这个结点的父结点就是我们要找的下一结点
      //不存在就说明没有下一节点
      if(pNode.next != null){
          TreeLinkNode pNext = pNode.next;
          while(pNext.next != null && pNext.next.right == pNext){
              //我们一直沿着父结点追朔，直到找到某个结点是其父结点的左子树
              pNext = pNext.next;
          }
          return pNext.next;
      }
      return null;
    }
}

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