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

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

---

 复杂的节点结构如下：

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

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

---

思路及解答:

1、通过循环，找到根节点，然后再通过根节点，中序遍历，在中序遍历的过程中，对比节点，是否等于输入的节点，然后获取下一个节点放回。注意：没有下一个节点的时候，应该返回null，不能数组越界。（借助list实现，空间复杂度为O(n)，时间上，遍历了两次，时间复杂度也是O(n)）

Java实现代码如下：

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

public class Solution{

    private static List<TreeLinkNode> treeLinkNodes = new ArrayList<>

        if(pNode != null){

        TreeLinkNode root = pNode;

        //一直找到根节点
        while(root != nul && root.next != null){

            root = root.next;
        }
    
        inOrder(root);
        for(int i = 0; i < treeLinkNodes.size(); i++){

            if(treeLinkNodes.get(i) == pNode){
                return i + 1 < treeLinkNodes.size() ? treeLinkNode.get(i + 1) : null;

            }
        }
    }

    //中序遍历
    public void inOrder(TreeLinkNode pNode){

        if(pNode != null){
            inOrder(pNode.left);
            treeLinkNodes.add(pNode);
            inOrder(pNode.right);
        }
    }
}

2、不借助额外的空间，直接查找下一个节点。分以下几种情况讨论：

• 当前节点为空，则直接返回空；
• 当前节点不为空：

如果当前节点的右节点不为空，那么下一个节点就是右节点的最左子孙节点；

如果当前节点的右节点为空，那么只能到父节点：

        需要判断当前节点是不是父节点的左节点，如果是父节点的左节点，那么下一个节点就是父节点。如果当前节点不是父节点的左节点，那么就是父节点的右节点，也就是下一个节点应该是父节点的父节点，或者更上一层。这个则需要根据当前节点是不是右节点来判断，如果是右节点，则还需要往父节点上走一层，如果不是右节点，则直接放回父节点。

Java代码实现如下：

public TreeLinkNode GetNext(TreeLinkNode pNode){

    //右节点不为空，直接找右节点的最左子孙节点
    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;
}

总结：通过以上练习，对于二叉树的结构及相关代码格式书写有了一些基础了解，同时也在书写过程中加深了对于二叉树及其遍历操作知识点的巩固，希望通过不断练习，能使用更简便的高效的方式编写代码来实现目标需求。