题目:
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
思路:
- 创建一个动态数组用于存储节点值、方法的返回
- 参数为空时返回空数组
- 创建一个队列用于存储未遍历的数组,并将根结点入队
- 记录当前处理的结点并控制台输出
- 将根结点存入要返回的数组
- 将当前结点的左右节点依次保存到队列的结尾(FIFO,先入先出)
代码:
package 从上往下打印二叉树23;
import java.util.ArrayList;
import java.util.LinkedList;
/**
* 从上往下打印出二叉树的每个节点,同层节点从左至右打印。
* @author Administrator
*/
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public class Demo {
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<>();
// 参数为空返回空数组
if (root == null) {
return list;
}
// 用于存储未遍历的元素,即当前结点的两个子结点
LinkedList<TreeNode> queue = new LinkedList<>();
// 将根结点入队
queue.add(root);
while (!queue.isEmpty()) {
// 用于记录当前处理的结点,poll:获取并移除队列的头
TreeNode treeNode = queue.poll();
// 输出队列头元素的值
System.out.print(treeNode.val + " ");
// 将根结点存入要返回的数组中
list.add(treeNode.val);
if (treeNode.left != null) {
// 将树的左子结点添加到此队列的结尾
queue.addLast(treeNode.left);
}
if (treeNode.right != null) {
// 将树的右子结点添加到此队列的结尾
queue.addLast(treeNode.right);
}
}
return list;
}
public static void main(String[] args) {
Demo demo = new Demo();
// 8
// / \
// 6 10
// / \ / \
// 5 7 9 11
TreeNode root = new TreeNode(8);
root.left = new TreeNode(6);
root.left.left = new TreeNode(5);
root.left.right = new TreeNode(7);
root.right = new TreeNode(10);
root.right.left = new TreeNode(9);
root.right.right = new TreeNode(11);
demo.PrintFromTopToBottom(root);
System.out.println("\n");
// 1
// /
// 3
// /
// 5
// /
// 7
// /
// 9
TreeNode root2 = new TreeNode(1);
root2.left = new TreeNode(3);
root2.left.left = new TreeNode(5);
root2.left.left.left = new TreeNode(7);
root2.left.left.left.left = new TreeNode(9);
demo.PrintFromTopToBottom(root2);
System.out.println("\n");
// 0
// \
// 2
// \
// 4
// \
// 6
// \
// 8
TreeNode root3 = new TreeNode(0);
root3.left = new TreeNode(2);
root3.left.left = new TreeNode(4);
root3.left.left.left = new TreeNode(6);
root3.left.left.left.left = new TreeNode(8);
demo.PrintFromTopToBottom(root3);
}
}