今日第二题:今天的题目是一道二叉树的题,它和“从上到下打印二叉树”十分类似,但是不需要多行,具体的解答见下面链接:
https://blog.csdn.net/qq_34428410/article/details/106453463
上题:
题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
分析:需要从上到下、从左到右打印二叉树,我们如果做过上面这个题,知道要用队列实现,但是这里有增加了一个条件,按层输出一行,即还需要一个队列存储下面的节点。
步骤:
1、建立两个Queue--one(存放当前结点)、two(存放孩子结点),用LinkedList,其实Queue的实现类。
2、还需要建立一个列表arr,存放每一层的节点值。
3、将当前要取值的节点存入one中;
4、遍历one节点,取出当前结点,将该节点的值放入arr中;
5、将one中当前结点的left和right放入two中;一致重复4-5,直到one中为空;
6、将arr存入二维列表中,one和two进行交换,使得子孩子变为根;
代码:
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
/*
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) {
Queue<TreeNode> one = new LinkedList<>();
Queue<TreeNode> two = new LinkedList<>();
one.offer(pRoot);
ArrayList<ArrayList<Integer>> array = new ArrayList<ArrayList<Integer>>();
if(pRoot==null)return array;
while(!one.isEmpty()){
ArrayList<Integer> arr = new ArrayList<>();
while(!one.isEmpty()){
TreeNode temp =one.poll();
arr.add(temp.val);
if(temp.left!=null){
two.offer(temp.left);
}
if(temp.right!=null){
two.offer(temp.right);
}
}
array.add(arr);
Queue<TreeNode> temp = one;
one=two;
two=temp;
}
return array;
}
}
牛客运行通过
运行时间:29ms
运行内存:9600Kb
其实和链接中的题目很像,只是增加了一个存储子孩子的队列,但是思想还是一样的。
欢迎各位互相交流学习哦~