剑指offer-59.按之字形顺序打印二叉树

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/zxm1306192988/article/details/82020413

牛客

题目描述
请实现一个函数按照之字形打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右至左的顺序打印，第三行按照从左到右的顺序打印，其他行以此类推。

题解：
在二叉树层次遍历的基础上修改，增设两个指针：

• p1 指向当前行的最右节点
• p2 向下一行的最右节点

遍历时，每当输出一个节点后，将把此节点的左右孩子压入队列，压入队列时，让 p2 跟踪入队的节点，然后判断 刚输出的节点是否是当前行的最右节点（node==p1？），如果是，就是说明当前的输出的结点是一行中最右结点，此时 p2 指向的节点一定是下一行的最右节点，将 p2 赋给 p1 。继续出栈，输出下一行。

import java.util.ArrayList;
import java.util.Collections;
import java.util.LinkedList;
import java.util.Queue;
public class Solution {
    public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        Queue<TreeNode> queue = new LinkedList();
        ArrayList<ArrayList<Integer>> result = new ArrayList<>();
        if (pRoot == null) {
            return result;
        }
        queue.add(pRoot);
        ArrayList<Integer> row = new ArrayList<Integer>();// 临时的list 装每一层的元素
        TreeNode p1 = pRoot; // 指向当前行的最右节点
        TreeNode p2 = pRoot; // 向下一行的最右节点
        Boolean flag = false;
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            row.add(node.val);
            if (node.left != null) {
                queue.add(node.left);
                p2 = node.left;
            }
            if (node.right != null) {
                queue.add(node.right);
                p2 = node.right;
            }
            if (node == p1) { // 输出的节点是否是当前行的最右节点
                if (flag) {
                    Collections.reverse(row);
                }
                result.add(row);
                row = new ArrayList<>();
                p1 = p2;
                if (flag) {
                    flag = false;
                } else {
                    flag = true;
                }
            }
        }
        return result;
    }

}