二叉树分层打印(美团面试题)

题目

给定一棵二叉树，返回 其节点值的层序遍历。（即从左到右，逐级）。

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

    3
   / \
  9  20
    /  \
   15   7
返回其级别顺序遍历：

[
  [3],
  [9,20],
  [15,7]
]

代码示例

代码不是完整的按照题目打印的 需要修改一下结果 能构造结果 本题也算是理解了

// 二叉树
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;
    }
}

public class Solution {
	
    public static void main(String[] args) {
        TreeNode root = getNode();
        if (root == null) {
            return;
        }
		//使用队列来放每层的元素
        Queue<TreeNode> queue = new LinkedList<>();
        //此处放第一层
        queue.add(root);
		
        while (!queue.isEmpty()) {
        	//有多少元素 循环多少次
            int levelSize = queue.size();

            for (int i = 0; i < levelSize; i++) {
            	// 这里会取出元素 
                TreeNode currentNode = queue.poll();
                System.out.print(currentNode.val + " ");
				// 如果有下一级的元素 则进行累加（如果是二层和三层的 可能是会共存的）
				// 跳出的判断在for循环进来的时候就确认了 所以 最终层级是分开的
                if (currentNode.left != null) {
                    queue.add(currentNode.left);
                }
                if (currentNode.right != null) {
                    queue.add(currentNode.right);
                }
            }
            System.out.println();
        }
    }

// 构造数据
    public static TreeNode getNode() {
        TreeNode t1 = new TreeNode(10);
        TreeNode t21 = new TreeNode(1);
        TreeNode t22 = new TreeNode(2);
        TreeNode t31 = new TreeNode(3);
        TreeNode t32 = new TreeNode(4);
        TreeNode t41 = new TreeNode(5);
        TreeNode t42 = new TreeNode(6);
        TreeNode t43 = new TreeNode(7);
        TreeNode t44 = new TreeNode(8);
        t32.left = t41;
        t32.right = t42;
        t31.left = t43;
        t31.right = t44;
        t22.left = t31;
        t22.right = t32;
        t1.left = t21;
        t1.right = t22;
        return t1;
    }
}

打印结果

10 
1 2 
3 4 
7 8 5 6