LeetCode 103：二叉树的锯齿形层次遍历

题目描述

给定一个二叉树，返回其节点值的锯齿形层次遍历。（即先从左往右，再从右往左进行下一层遍历，以此类推，层与层之间交替进行）。

例如：
给定二叉树 [3,9,20,null,null,15,7],

3

/
9 20
/
15 7
返回锯齿形层次遍历如下：

[
[3],
[20,9],
[15,7]
]

解题思路

二叉树的锯齿形遍历，顾名思义，奇数层与偶数层遍历顺序相反。按题目要求，奇数层从左到右，偶数层从右到左打印。我们可以借助栈的结构，然后用一个flag作为奇偶数的标志位，一个栈用来存放奇数层节点，一个栈用来存放偶数层节点。当为奇数层时，先从栈中弹出节点，添加到arraylist当中，因为下一次为偶数层，需要从右至左打印，所以要先存放当前奇数栈中弹出节点的左子树节点，再存放其右子树节点（存到偶数层栈中）！到偶数层时，先弹出当前栈的顶部节点，添加到arraylist中，然后依次存放其右子树节点和左子树节点到奇数层栈中！

代码实现

去博客设置页面，选择一款你喜欢的代码片高亮样式，下面展示同样高亮的 代码片.

import java.util.ArrayList;
import java.util.Stack;

public class s_printOfTreeNode {
    //将二叉树的节点按照之字形打印
    public ArrayList<ArrayList<Integer>> printNode(TreeNode root) { //acc:1ms 99%
        ArrayList<ArrayList<Integer>> res = new ArrayList<>();
        Stack<TreeNode> s1 = new Stack<>();//存放偶数层节点
        Stack<TreeNode> s2 = new Stack<>();//存放奇数层节点
        int flag = 1;//标志位 默认从第一层（奇数层）开始
        if (root == null) { //二叉树为空
            return res;
        }
        s2.push(root); //存放根节点
        ArrayList<Integer> list = new ArrayList<>();//存放每层打印的结果
        while (!s1.isEmpty() || !s2.isEmpty()) {
            if (flag % 2 != 0) {//奇数层
                while (!s2.isEmpty()) {
                    TreeNode node = s2.pop();
                    list.add(node.val);
                    if (node.left != null) {
                        s1.push(node.left);
                    }
                    if (node.right != null) {
                        s1.push(node.right);
                    }

                }

            }
            if (flag % 2 == 0) {//偶数层
                while (!s1.isEmpty()) {
                    TreeNode node = s1.pop();
                    list.add(node.val);
                    if (node.right != null) {
                        s2.push(node.right);
                    }
                    if (node.left != null) {
                        s2.push(node.left);
                    }
                }

            }
            res.add(list);
            list = new ArrayList<Integer>();
            flag++;

        }
        return res;
    }


    public static void main(String[] args) {
        TreeNode node1 = new TreeNode(10);
        TreeNode node2 = new TreeNode(6);
        TreeNode node3 = new TreeNode(14);
        TreeNode node4 = new TreeNode(4);
        TreeNode node5 = new TreeNode(8);

        node1.left = node2;
        node1.right = node3;
        node2.left = node4;
        node2.right = node5;
        node3.left = null;
        node3.right = null;
        node4.left = null;
        node4.right = null;
        node5.left = null;
        node5.right = null;

        s_printOfTreeNode test = new s_printOfTreeNode();
        ArrayList<ArrayList<Integer>> res = test.printNode(node1);
        System.out.println(res);
    }
}

总结

本题来源于Leetcode中 归属于栈类型题目。
同许多在算法道路上不断前行的人一样，不断练习，修炼自己！
如有博客中存在的疑问或者建议，可以在下方留言一起交流，感谢各位！

觉得本博客有用的客官，可以给个赞鼓励下！ 嘿嘿