题目描述
从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。
把二叉树打印成多行
解题思路
- 使用一个ArrayList<ArrayList<Integer>>保存各层次节点
- Queue<TreeNode> 暂存下一层节点
循环部分：
- 使用变量levelNum 暂存queue当前层的节点数目
- 每次获取queue队列顶部的节点，从左到右遍历
- 同时使用tmpList保存每一层的数据，添加到resultList中
- 最后返回结果
代码实现
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Solution {
ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> resultList = new ArrayList<>();// 存储多层数据
Queue<TreeNode> queue = new LinkedList<>(); // 暂存下一层节点
if (pRoot == null) {
return resultList;
}
int levelNum = 0;
queue.add(pRoot);
while (!queue.isEmpty()){
levelNum = queue.size(); // 当前层节点个数
ArrayList<Integer> tmpList = new ArrayList<>();
while (levelNum > 0){ // 在queue中遍历当前层的所有节点
TreeNode tmpNode = queue.remove(); // 弹出顶层节点
if (tmpNode != null){ // 节点不为空，添加其左右节点，如果其左右节点为null，在下一层当中不会添加到queue队列中
tmpList.add(tmpNode.val);
queue.add(tmpNode.left);
queue.add(tmpNode.right);
}
levelNum--;
}
if (tmpList.size() > 0){
resultList.add(tmpList);
}
}
return resultList;
}
}
本文由博客群发一文多发等运营工具平台 OpenWrite 发布
