剑指offer面试题8：二叉树的下一个节点（两种递归解法）

题目描述

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

由于作者的递归思想掌握的不好，故本题尝试用递归方法求解。

第一种递归方法

画一个二叉树，然后模拟一下，可知用递归来实现是比较方便的。
并且还有两个特点：
① 当前节点单独考虑，因为找的下一个节点必然是右子树的最左节点（右子树存在的条件下）；
② 右子树为空需要单独考虑，因为右子树可能是整棵树的最右节点，故而没有下一个节点，也有可能是较好条件下的节点，找到出现左路径的直接根节点即可；

/*
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.right == null) {//分两种情况
			return getFather(pNode);
		}else {
        	return findTreeNode(pNode.right);
        }
    }

	private TreeLinkNode getFather(TreeLinkNode pNode) {//当右子节点为空时，向上递归
        if(pNode.next == null) return null;
		if(pNode.next.right == pNode) return getFather(pNode.next);
		else {//直到找到pNode的父节点的左子树为pNode
			return pNode.next;
		}
	}

	private TreeLinkNode findTreeNode(TreeLinkNode node) {//当右子节点不为空时，向下递归
		
        if(node.left==null) return node;
		else {
			return findTreeNode(node.left);
		}
	}
}

第二章递归方法

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

/*
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 {
    static ArrayList<TreeLinkNode> list = new ArrayList<>();
    public TreeLinkNode GetNext(TreeLinkNode pNode){
        TreeLinkNode par = pNode;
        while(par.next != null){
            par = par.next;
        }
        InOrder(par);
        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;
    }
    void InOrder(TreeLinkNode pNode){//建立二叉树中序遍历序列
        if(pNode!=null){
            InOrder(pNode.left);
            list.add(pNode);
            InOrder(pNode.right);
        }
    }
}