【算法】解题总结：剑指Offer 57 二叉树的下一个结点、剑指Offer 60 把二叉树打印成多行

JZ57 二叉树的下一个结点

（中等）

题目

描述
给定一个二叉树其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。注意，树中的结点不仅包含左右子结点，同时包含指向父结点的next指针。下图为一棵有9个节点的二叉树。树中从父节点指向子节点的指针用实线表示，从子节点指向父节点的用虚线表示

示例:
输入:{8,6,10,5,7,9,11},8
返回:9
解析:这个组装传入的子树根节点，其实就是整颗树，中序遍历{5,6,7,8,9,10,11}，根节点8的下一个节点就是9，应该返回{9,10,11}，后台只打印子树的下一个节点，所以只会打印9，如下图，其实都有指向左右孩子的指针，还有指向父节点的指针，下图没有画出来

输入描述：
输入分为2段，第一段是整体的二叉树，第二段是给定二叉树节点的值，后台会将这2个参数组装为一个二叉树局部的子树传入到函数GetNext里面，用户得到的输入只有一个子树根节点
返回值描述：
返回传入的子树根节点的下一个节点，后台会打印输出这个节点

示例1
输入：
{8,6,10,5,7,9,11},8
返回值：
9

示例2
输入：
{8,6,10,5,7,9,11},6
返回值：
7

示例3
输入：
{1,2,#,#,3,#,4},4
返回值：
1

示例4
输入：
{5},5
返回值：
“null”
说明：
不存在，后台打印"null"

思路

此题用暴力解法也可以做，因为已经有指向父结点的 next 属性了，因此可以追溯到根结点，进行并记录一次中序遍历，我们只需将传入的参数结点的引用与记录遍历结构的链表中的结点进行对比，找出相等的结点的下一个即可，此方法十分容易实现，我当时在想到这种方法时，又接着想到了下面这种按右子树来划分判断的方法，因此就不打算用这种暴力无脑的方法了，下面是详细分析：

此题考察我们对树结构中序遍历的熟悉情况，中序遍历的递归思想是：左递归; print; 右递归; ，因此，此题要找出中序遍历时传入的参数结点的下一个结点就比较好想了，根据我们对树结构的经验，此题在 print 之后去执行的操作就是一个突破条件，即右递归操作，即对应二叉树中的右子树。

如果传入的结点，有右子树，那么中序遍历的下一个结点一定是右子树中的某一个结点，如果没有右子树，那么则一定是之前的某个祖先结点。

因此，更详细地来讲：

当参数结点 ：

• 有右子树时：中序遍历的下一个结点一定是右子树中的某一个结点，让一个临时结点 tmp 指向此右子树的根节点，即右孩子，一直像左延伸，知道不再此 tmp 结点不再有左孩子，那么根据 左递归; print; 右递归 可知，下一个输出的一定是之前参数结点的右孩子；
• 当没有右子树时：中序遍历的下一个结点可能是之前的某个祖先结点（特殊情况就是也可能此结点就是中序遍历的最后一个结点），则一直向上追溯至符合某结点是其父结点的左孩子的结点对应的父结点即可，也要注意不要溢出根节点（听起来很绕，但我没想到更好的描述方式）。

再把上面的图拉下来帮助理解：

实现

public class JZ57二叉树的下一个结点 {

    public TreeLinkNode GetNext(TreeLinkNode pNode) {
        if (pNode == null) {
            return null;
        }

        TreeLinkNode tmp = pNode;

        if (tmp.right != null) { //此结点 有右子树
            tmp = tmp.right; //下一个结点在右子树

            while (tmp.left != null) {
                tmp = tmp.left;
            }
            return tmp;

        } else { //此结点 没有右子树
            //则一直向上追溯至符合某结点是其父结点的左孩子 的父结点
            while (tmp.next.left != tmp) {
                //一直向上追溯需注意到根节点的问题
                if (tmp.next == null) {
                    return null;
                }
                tmp = tmp.next;
            }
            return tmp.next;
        }

    }
}

JZ60 把二叉树打印成多行

（中等）

题目

描述
从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。
例如：
给定的二叉树是{1,2,3,#,#,4,5}

该二叉树多行打印层序遍历的结果是
[
[1],
[2,3],
[4,5]
]

示例1
输入：
{1,2,3,#,#,4,5}
返回值：
[[1],[2,3],[4,5]]

示例2
输入：
{8,6,10,5,7,9,11}
返回值：
[[8],[6,10],[5,7,9,11]]

示例3
输入：
{1,2,3,4,5}
返回值：
[[1],[2,3],[4,5]]

思路

此题和 JZ59 按之字形顺序打印二叉树 这道题几乎一样，而且相比于那道题更加简单，此题只需要每层都从左到右记录结果即可，具体解析可直接参考之前文章：【算法面试题系列】详细解析：按之字形顺序打印二叉树、二叉搜索树的第k个结点

实现

public class JZ60把二叉树打印成多行 {
    ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> deepList = new ArrayList<>();

        if (pRoot == null) {
            return deepList;
        }
        Queue<TreeNode> queue = new LinkedList();
        queue.add(pRoot);
        while (!queue.isEmpty()) {
            int size = queue.size(); //此行的结点个数
            ArrayList<Integer> list = new ArrayList<>();
            while (size > 0) {
                TreeNode node = queue.poll();
                size--;
                list.add(node.val);
                if (node.left != null) queue.add(node.left);
                if (node.right != null) queue.add(node.right);
            }
            deepList.add(list);
        }
        return deepList;
    }
}