zl程序教程

您现在的位置是:首页 >  后端

当前栏目

剑指offer解法汇总77-按之字形顺序打印二叉树

二叉树 汇总 打印 顺序 Offer 解法 77
2023-09-11 14:18:52 时间

原题链接:按之字形顺序打印二叉树_牛客题霸_牛客网

描述:

给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)

数据范围:0≤n≤1500,树上每个节点的val满足 |val| <= 100∣val∣<=100
要求:空间复杂度:O(n)O(n),时间复杂度:O(n)O(n)

例如:
给定的二叉树是{1,2,3,#,#,4,5}

该二叉树之字形层序遍历的结果是

[

[1],

[3,2],

[4,5]

]

示例1

输入:

{1,2,3,#,#,4,5}

返回值:

[[1],[3,2],[4,5]]

说明:

如题面解释,第一层是根节点,从左到右打印结果,第二层从右到左,第三层从左到右。

解题思路:

一层一层的遍历

代码:

public ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        if (pRoot == null) {
            return list;
        }
        ArrayList<TreeNode> inList = new ArrayList<>();
        inList.add(pRoot);
        ArrayList<TreeNode> outList = new ArrayList<>();
        boolean isFromLeft = true;
        while (inList.size() > 0) {
            outList.clear();
            ArrayList<Integer> integers = PrintLevel(isFromLeft, inList, outList);
            inList.clear();
            inList.addAll(outList);
            list.add(integers);
            isFromLeft = !isFromLeft;
        }
        return list;
    }

    public ArrayList<Integer> PrintLevel(boolean isFromLeft, ArrayList<TreeNode> inList, ArrayList<TreeNode> outList) {
        ArrayList<Integer> list = new ArrayList<>();
        for (int i = 0; i < inList.size(); i++) {
            TreeNode node = inList.get(i);
            if (isFromLeft) {
                list.add(node.val);
            } else {
                list.add(0, node.val);
            }
            if (node.left != null) {
                outList.add(node.left);
            }
            if (node.right != null) {
                outList.add(node.right);
            }
        }
        return list;
    }