面试必刷TOP101:27、按之字形顺序打印二叉树
  9ShvDtAiXXil 2023年11月15日 33 0

题目

面试必刷TOP101:27、按之字形顺序打印二叉树_i++

题解

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param pRoot TreeNode类
     * @return int整型ArrayList<ArrayList<>>
     */
    public ArrayList<ArrayList<Integer>> Print (TreeNode pRoot) {
        // write code here
        ArrayList<ArrayList<Integer>> layers = new ArrayList<ArrayList<Integer>>();

        if (pRoot == null) {
            return layers;
        }

        ArrayList<Integer> layer = new ArrayList<Integer>();
        Queue<TreeNode> queue = new LinkedList<TreeNode>();

        // Initialize first node
        layer.add(pRoot.val);
        layers.add(layer);
        queue.offer(pRoot);
        boolean reverse = false;

        while (!queue.isEmpty()) {
            // Reverse flag
            reverse = !reverse;
            layer = new ArrayList<Integer>();

            Stack<Integer> stack = new Stack<Integer>();
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode node = queue.poll();

                // Loop for `node.left` and `node.right`
                for (int j = 0; j < 2; j++) {
                    TreeNode current = j < 1 ? node.left : node.right;
                    if (current != null) {
                        // System.out.println(
                        //     node.val + (j < 1 ? ".left = " : ".right = ") + current.val
                        // );

                        if (!reverse) {
                            // Append to result in ascending order
                            layer.add(current.val);
                        } else {
                            // Reverse order for current layer
                            stack.push(current.val);
                        }

                        // Add node to queue for next layer
                        queue.offer(current);
                    }
                }
            }

            // Reverse needed for the current layer
            while (!stack.isEmpty()) {
                layer.add(stack.pop());
            }

            if (layer.size() > 0) {
                layers.add(layer);
            }
        }

        return layers;
    }
}
【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月15日 0

暂无评论

推荐阅读
  2Vtxr3XfwhHq   2024年05月17日   53   0   0 Java
  Tnh5bgG19sRf   2024年05月20日   107   0   0 Java
  8s1LUHPryisj   2024年05月17日   46   0   0 Java
  aRSRdgycpgWt   2024年05月17日   47   0   0 Java
9ShvDtAiXXil