剑指Offer第二十二题:层序遍历二叉树
层序遍历需要使用一个队列来保存每一层每一个结点,当队列中一个结点出队时,将其左右子结点入队。
import java.util.*;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
//层序遍历二叉树
public class Solution {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> res = new ArrayList<Integer>();
if(root == null){
return res;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
TreeNode tempNode;
while(!queue.isEmpty()){
tempNode = queue.poll();
if(tempNode.left != null){
queue.add(tempNode.left);
}
if(tempNode.right != null){
queue.add(tempNode.right);
}
res.add(tempNode.val);
}
return res;
}
}
这道题的升级版
要求按行输出每一层的结点,一层输出结束后要换行
- 使用两个结点分别来标记当前行最右边的结点和下一行最右边的结点
- 继续使用一个临时变量来指向当前出队结点
- 每一次一个结点入队,就将下一行最右结点更新
- 当临时结点和当前行最右结点相等时,说明需要换行了,换行并将当前行最右结点更新为下一行最右结点
import java.util.LinkedList;
import java.util.Queue;
public class 按行输出层序遍历二叉树 {
public static void printTree(TreeNode root) {
if(root == null) {
return;
}
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
TreeNode lastNode1 = root;//标记当前行最右结点,当临时结点等于当前行最右结点时,当前行遍历结束,换行,
//并将lastNode1和lastNode2更新
TreeNode lastNode2 = root;//标记下一行最右结点,每添加一个结点都会更新
TreeNode tempNode;//临时结点
while(!queue.isEmpty()) {
tempNode = queue.poll();
System.out.print(tempNode.val+" ");
if(tempNode.left != null) {
queue.add(tempNode.left);
lastNode2 = tempNode.left;
}
if(tempNode.right != null) {
queue.add(tempNode.right);
lastNode2 = tempNode.right;
}
if(lastNode1 == tempNode) {
System.out.println();
lastNode1 = lastNode2;
}
}
}
}