剑指offer 32 从上到下打印二叉树

最新推荐文章于 2023-03-21 17:05:00 发布

原创最新推荐文章于 2023-03-21 17:05:00 发布 · 228 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #java #队列 #广度优先搜索

八股编程专栏收录该内容

86 篇文章

订阅专栏

本文介绍了一种从上到下、从左至右的二叉树广度优先遍历算法，通过使用队列实现，适用于节点总数不超过1000的二叉树。该算法在LeetCode上表现优异，执行效率高。

@author: sdubrz
@date:  8/2/2020 9:24:21 AM  
难度： 中等
考察内容： 二叉树
@e-mail: lwyz521604#163.com
题目来自《剑指offer》 电子工业出版社

从上到下打印出二叉树的每个节点，同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树: [3,9,20,null,null,15,7],

[3,9,20,15,7]

提示：

节点总数 <= 1000

解法

这个问题其实是一个广度优先搜索，可以用一个队列来实现：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public int[] levelOrder(TreeNode root) {
        if(root==null) {
			return new int[0];
		}
		Queue<TreeNode> queue = new LinkedList<>();
		ArrayList<Integer> list = new ArrayList<>();
		queue.add(root);
		
		while(!queue.isEmpty()) {
			TreeNode current = queue.poll();
			list.add(current.val);
			if(current.left!=null) {
				queue.add(current.left);
			}
			if(current.right!=null) {
				queue.add(current.right);
			}
		}
		
		int[] result = new int[list.size()];
		Iterator<Integer> iter = list.iterator();
		int index = 0;
		while(iter.hasNext()) {
			result[index] = (int) iter.next();
			index++;
		}
		return result;
    }
}

在 LeetCode 系统中提交的结果为

执行结果：通过 显示详情
执行用时：1 ms, 在所有 Java 提交中击败了99.65%的用户
内存消耗：40 MB, 在所有 Java 提交中击败了35.23%的用户