57_GetNext

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

   思路1：需要仔细分析找规律

    分析二叉树中序遍历的下一个节点，一共有以下情况：
    1.二叉树为空，则返回空；
    2.节点右孩子存在，则设置一个指针从该节点的右孩子出发，一直沿着指向左子结点的指针找到的叶子节点即为下一个节点；
    3.节点不是根节点，右孩子不存在。如果该节点是其父节点的左孩子，则返回父节点；为右孩子，
      则继续向上遍历其父节点的父节点，直到找到第一个结点cur，该结点的父节点root的左子节点为root.left=cur，返回root。

 

 

 

   思路2：挺好的思路。O(n)
    1）从pNode一直追溯到根结点
    2）从根结点开始中序遍历，将遍历结果加入到集合中
    3）遍历集合，顺序找到pNode，返回pNode下一个结点

 

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Solution {


    public static void main(String[] args) {
        TreeLinkNode n1 = new TreeLinkNode(1);
        TreeLinkNode n2 = new TreeLinkNode(2);
        TreeLinkNode n3 = new TreeLinkNode(3);
        TreeLinkNode n4 = new TreeLinkNode(4);
        TreeLinkNode n5 = new TreeLinkNode(5);
        n1.left = n2;
        n1.right = n3;
        n2.left = n4;
        n2.right = n5;
        n2.next = n1;
        n4.next = n2;
        n5.next = n2;
        n3.next = n1;
        Solution s = new Solution();
        TreeLinkNode res = s.GetNext2(n4);
        System.out.println(res.val);
    }

    TreeLinkNode pre = null;

    public TreeLinkNode GetNext(TreeLinkNode pNode) {
        if (pNode == null)
            return null;
        if (pNode.right != null) {//如果有右子树，则找右子树的最左节点
            pNode = pNode.right;
            while (pNode.left != null) {
                pNode = pNode.left;
            }
            return pNode;
        }
        while (pNode.next != null) {//没右子树，则找第一个当前节点是父节点左孩子的节点
            TreeLinkNode proot = pNode.next;
            if (proot.left == pNode)
                return proot;//返回父节点
            pNode = pNode.next;
        }
        return null;//退到了根节点仍没找到，则返回null
    }

    public TreeLinkNode GetNext2(TreeLinkNode pNode) {
        TreeLinkNode cur = pNode;//注意如果直接使用pNode，最后pNode将指向root，所以要保存pNode
        if (cur == null)
            return null;
        while (cur.next != null) {
            cur = cur.next;
        }
        ArrayList<TreeLinkNode> nodesList = new ArrayList<>();
        midOrder(cur, nodesList);//将中序遍历的结果加入集合
        for (int i = 0; i < nodesList.size(); i++) {
            //要判断是否是最后一个，否则会空指针异常
            if (nodesList.get(i) == pNode && i != nodesList.size() - 1) {
                return nodesList.get(i + 1);
            }
        }

        return null;
    }

    public void midOrder(TreeLinkNode root, List<TreeLinkNode> list) {
        if (root == null)
            return;
        midOrder(root.left, list);
        list.add(root);
        midOrder(root.right, list);

    }


}

class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}