从今天开始刷算法题-2021.12.06(补)

剑指 Offer 32 - I. 从上到下打印二叉树

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-lcof/

1. 使用队列存储每一层的元素
2. 每次遍历队列中的节点的值加入list中，将节点的左节点入队，右节点入队
3. 遍历list并输出int[]

• 代码如下:

  class Solution {
      public int[] levelOrder(TreeNode root) {
          if(root==null) return new int [0];
          Queue<TreeNode> queue = new LinkedList<>();
          queue.add(root);
          ArrayList<Integer> ans = new ArrayList<>();
          while(!queue.isEmpty()){
              TreeNode node = queue.poll();
              ans.add(node.val);
              if(node.left!=null) queue.add(node.left);
              if(node.right!=null) queue.add(node.right);
          }
          int[] res = new int[ans.size()];
          for(int i=0;i<ans.size();i++){
              res[i] = ans.get(i);
          }
          return res;
      }
  }

从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

4. 在上述基础上，每次遍历队列时，将队列中的元素都写到一个新的list中

• 代码如下：

  class Solution {
      public List<List<Integer>> levelOrder(TreeNode root) {
          Queue<TreeNode> queue = new LinkedList<>();
          List<List<Integer>> res = new ArrayList<>();
          if(root != null) queue.add(root);
          while(!queue.isEmpty()) {
              List<Integer> tmp = new ArrayList<>();
              for(int i = queue.size(); i > 0; i--) {
                  TreeNode node = queue.poll();
                  tmp.add(node.val);
                  if(node.left != null) queue.add(node.left);
                  if(node.right != null) queue.add(node.right);
              }
              res.add(tmp);
          }
          return res;
      }
  }

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

5. 在4.的基础上，对偶数层倒序处理

• 代码如下：

  class Solution {
      public List<List<Integer>> levelOrder(TreeNode root) {
          Queue<TreeNode> queue = new LinkedList<>();
          List<List<Integer>> res = new ArrayList<>();
          if(root != null) queue.add(root);
          while(!queue.isEmpty()) {
              List<Integer> tmp = new ArrayList<>();
              for(int i = queue.size(); i > 0; i--) {
                  TreeNode node = queue.poll();
                  tmp.add(node.val);
                  if(node.left != null) queue.add(node.left);
                  if(node.right != null) queue.add(node.right);
              }
              if(res.size() % 2 == 1) Collections.reverse(tmp);
              res.add(tmp);
          }
          return res;
      }
  }