二叉树特点:叶子节点最多两个,分为左叶子节点和右叶子节点。
思路:将每一层节点放入list集合或者队列,遍历完当前层再继续遍历下一层。
class TreeNode {
int val;
TreeNode leftNode;
TreeNode rightNode;
TreeNode(int val) {
this.val = val;
}
}
public class TreeTest {
public static void main(String[] args) {
TreeNode root =new TreeNode(5);
TreeNode leftNode=new TreeNode(3);
TreeNode leftNodeL=new TreeNode(2);
TreeNode leftNodeR=new TreeNode(1);
TreeNode rightNode=new TreeNode(6);
TreeNode rightNodeL=new TreeNode(7);
TreeNode rightNodeR=new TreeNode(8);
leftNode.leftNode=leftNodeL;
leftNode.rightNode=leftNodeR;
rightNode.rightNode=rightNodeR;
rightNode.leftNode=rightNodeL;
root.leftNode=leftNode;
root.rightNode=rightNode;
levelPrintln(root);
}
public static void levelPrintln(TreeNode root){
LinkedList<TreeNode> linkedList=new LinkedList<>();
linkedList.offer(root);
while(!linkedList.isEmpty()){
ArrayList<Integer> arrayList=new ArrayList<>();
int size=linkedList.size();
while(size>0){
TreeNode treeNode=linkedList.poll();
arrayList.add(treeNode.val);
if(treeNode.leftNode!=null){
linkedList.offer(treeNode.leftNode);
}
if(treeNode.rightNode!=null){
linkedList.offer(treeNode.rightNode);
}
size --;
}
System.out.println(arrayList);
}
思考:二叉树按层打印,第一反应是BFS(广度优先搜索),如何按层将每一层打印完,再打印下一层这是重点。
代码中使用size()方法,将每层元素取出来放入一个新的list集合。size=0时表示该层遍历结束。
直接出打印新的集合即可。
如何将二叉树倒着按层打印?
最简单的办法将每层集合放入一个新的集合,比如前边说过的栈结构,放入栈中遍历调用pool()打印。