题解 | #按之字形顺序打印二叉树#

按之字形顺序打印二叉树

https://www.nowcoder.com/practice/91b69814117f4e8097390d107d2efbe0

import java.util.ArrayList;
import java.util.LinkedList;

public class Solution {
    public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> list = new ArrayList<>();
        if (pRoot == null) return list;
        LinkedList<TreeNode> queue = new LinkedList<>();
        int levelflag = 1;
        queue.offer(pRoot);
        while (!queue.isEmpty()) {
            ArrayList<Integer> level = new ArrayList<>();
            LinkedList<TreeNode> temp = new LinkedList<>();
            temp.addAll(queue);
            int num = queue.size();
            if (levelflag % 2 == 0) {
                for (int i = 0 ; i < num; i++) {
                    TreeNode p = temp.getLast();
                    if (p.right != null) {
                        queue.offer(p.right);
                    }
                    if (p.left != null) {
                        queue.offer(p.left);
                    }
                    level.add(p.val);
                    queue.remove(p);
                    temp.removeLast();
                }
            } else {
                for (int i = 0 ; i < num; i++) {
                    TreeNode p = temp.getLast();
                    if (p.left != null) {
                        queue.offer(p.left);
                    }
                    if (p.right != null) {
                        queue.offer(p.right);
                    }
                    level.add(p.val);
                    queue.remove(p);
                    temp.removeLast();
                }
            }
            list.add(level);
            levelflag++;
        }
        return list;
    }
}

全部评论

相关推荐

2024-12-20 21:43
湖北大学 Java
黑皮白袜臭脚体育生:项目加一个毛遂自荐一下,开源仿b站微服务项目,GitHub已经390star,牛客上有完整文档教程,如果觉得有帮助的话可以点个小星星,蟹蟹
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务