从上到下打印二叉树(Java)
题目:如果一个二叉树的先序遍历为 1 2 4 5 3 6 7 则从上到下打印为 1 2 3 4 5 6 7 (先序遍历为先打印根节点,在左节点,在右节点)。
代码:
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.Queue;
/**
* @author zhaoliang
* @desc 从上到下打印二叉树
* @date 20200507
*/
public class Main22 {
//从上往下打印出二叉树的每个节点,同层节点从左至右打印。
public int[] levelOrder(TreeNode root) {
if (root == null)
return new int[0];
ArrayList<Integer> ans = new ArrayList<>();
Queue<TreeNode> queue = new LinkedList<>();
queue.add(root);
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
ans.add(node.val);
if (node.left != null)
queue.add(node.left);
if (node.right != null)
queue.add(node.right);
}
int[] a = new int[ans.size()];
for (int i = 0; i < ans.size(); i++) {
a[i] = ans.get(i);
}
return a;
}
}