1.题目
给定一个节点数为 n 二叉树,要求从上到下按层打印二叉树的 val 值,同一层结点从左至右输出,每一层输出一行,将输出的结果存放到一个二维数组中返回。
2.解法
2.1 树的结构
public 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) {
}
}
2.2 解法1
使用宽度优先队列,Queue<TreeNode> queue = new ArrayDeque()
。按层级将节点输出,并将子节点放入队列。
ArrayList<ArrayList<Integer>> Print(TreeNode pRoot) {
ArrayList<ArrayList<Integer>> ans = new ArrayList<>();
Queue<TreeNode> queue = new ArrayDeque();
if (pRoot == null) return ans;
queue.offer(pRoot);
while (!queue.isEmpty()) {
int size = queue.size();
ArrayList<Integer> curlevel = new ArrayList<>();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
curlevel.add(node.val);
if (node.left != null) queue.offer(node.left);
if (node.right != null) queue.offer(node.right);
}
ans.add(curlevel);
}
return ans;
}
总结
使用宽度优先队列,Queue<TreeNode> queue = new ArrayDeque()
。按层级将节点输出,并将子节点放入队列。
算法系列在github上有一个开源项目,主要是本系列博客的demo代码。https://github.com/forestnlp/alg
如果您对软件开发、机器学习、深度学习有兴趣请关注本博客,将持续推出Java、软件架构、深度学习相关专栏。
您的支持是对我最大的鼓励。