二叉树按层遍历

题目

如下图所示,给定一棵二叉树的头节点head,按层打印二叉树,打印结果如下所示:


1
2 3
4 5 6
7 8

分析

定义两个变量:
last: 表示正在打印的当前行的最右节点
nlast:表示下一行的最右节点

定义一个队列 queue ,首先让 last 指向头节点 1;然后将 1 存入队列 queue,弹出 1,并打印 1,往队列中放入 1 的结点的左孩子 2 并让 nlast 指向节点 2,往队列中放入 1 的右孩子 3 并让 nlast 指向节点3;
此时发现弹出的结点 1 与 last 相等,所以该换行了,令 last = nlast ,然后从队列中弹出 2 并打印,把结点2 的孩子存入队列,并令 nlast指向节点2的孩子,弹出节点 3 并打印,将节点3的孩子添加到队列,并令nlast指向节点3的孩子, 此时发现弹出的节点 3 就是last 节点,所以该换行了。以此类推

代码实现

import java.util.*;

/**
 * 有一棵二叉树,请设计一个算法,按照层次打印这棵二叉树。
 * 给定二叉树的根结点root,请返回打印结果,结果按照每一层一个数组进行储存,
 * 所有数组的顺序按照层数从上往下,且每一层的数组内元素按照从左往右排列。
 * 保证结点数小于等于500。
 */
public class EX_01 {
    private class TreeNode {
        int val = 0;
        TreeNode left = null;
        TreeNode right = null;
        public TreeNode(int val) {
            this.val = val;
        }
    }

    public int[][] printTree(TreeNode root) {
        ArrayList<ArrayList<Integer>> rows = new ArrayList<>();
        ArrayList<Integer> row = new ArrayList<>();
        TreeNode last = root;//记录正在打印行的最后一个节点
        TreeNode nlast = null;//记录下一行的最后一个节点
        TreeNode temp = null;
        ArrayDeque<TreeNode> queue = new ArrayDeque<>();
        queue.add(root);//将根节点添加至队列
        while (!queue.isEmpty()) {
            temp = queue.poll();//获取队头元素
            row.add(temp.val);
            if (temp.left != null) {//添加左孩子并使nlast指向此处
                queue.add(temp.left);
                nlast = temp.left;
            }
            if (temp.right != null) {//添加右孩子并使nlast指向此处
                queue.add(temp.right);
                nlast = temp.right;
            }
            if (temp == last) {//判断当前行是否打印完毕
                last = nlast;
                rows.add(row);//将一行数据添加到结果中
                row = new ArrayList<>();//新建一行
            }
        }
        int[][] res = new int[rows.size()][];//结果转化成数组形式
        for (int i = 0; i < rows.size(); i++) {
            res[i] = new int[rows.get(i).size()];
            for (int j = 0; j < res[i].length; j++) {
                res[i][j] = rows.get(i).get(j);
            }
        }
        return res;
    }
}






  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值