返回介绍

59.按之字形打印二叉树

发布于 2023-08-30 21:54:39 字数 975 浏览 0 评论 0 收藏 0

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

思路:BFS,使用队列,用一个标志位来决定是否需要翻转。

import java.util.*;
public class Solution {
  public ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
    ArrayList<ArrayList<Integer> > result = new ArrayList<>();
    if(pRoot == null) return result;
    Queue<TreeNode> queue = new LinkedList<>();
    queue.add(pRoot);
    boolean reverse = false;
    while(!queue.isEmpty()){
      ArrayList<Integer> list = new ArrayList<>();
      int size = queue.size();
      for(int i = 0; i < size; i++){
        TreeNode node = queue.poll();
        list.add(node.val);
        if(node.left != null) queue.add(node.left);
        if(node.right != null) queue.add(node.right);
      }
      if(reverse){
        Collections.reverse(list);
      }
      result.add(list);
      reverse = !reverse;
    }
    return result;
  }
}

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
    我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
    原文