二叉树之层序遍历

cszdlt

已于 2024-01-04 19:49:43 修改

阅读量475

点赞数 8

文章标签：数据结构算法

于 2024-01-04 19:47:11 首次发布

本文链接：https://blog.csdn.net/cszdlt/article/details/135394520

版权

本文介绍了二叉树的层序遍历算法，包括遍历流程图和代码示例，并针对LeetCode116题，演示如何在遍历过程中将左节点的next指针指向右侧节点。

摘要由CSDN通过智能技术生成

前言：

二叉树的横向遍历是指把二叉树按照从上到下依次进行遍历，每一层具体是从左到右还是从右到左遍历都是可以根据题目要求进行实现的。二叉树的层序遍历流程图以及代码如下图所示。

层序遍历流程图：

图中画的每层的遍历方向是从左向右进行遍历的。为了方便理解，图中画的二叉树为一颗满二叉树，这个对于层序遍历来说没有有求。

层序遍历代码:

public List<Integer> layerForEach(TreeNode root) {
        List<Integer> result = new ArrayList<>();
        if (root == null) {
            return result;
        }
        Queue<TreeNode> curLayer = new LinkedList<>();
        curLayer.offer(root);
        while (!curLayer.isEmpty()) {
            int curSize = curLayer.size();
            for (int i = 0; i < curSize; i++) {
                TreeNode poll = curLayer.poll();
                if (poll.left != null) {
                    curLayer.add(poll.left);
                }
                if (poll.right != null) {
                    curLayer.add(poll.right);
                }
                result.add(poll.val);
            }
        }
        return result;
}

层序遍历练习:

以leetcode116题为例，通过题意可以get到，题目是想让我们按层序遍历一颗二叉树，然后把左节点的next指针指向右侧节点（不一定是其父节点的右孩子）

解决代码：

public Node connect(Node root) {
        Queue<Node> queue = new LinkedList<>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            int size = queue.size();
            // 单独处理每一层的第一个接点
            Node prePoll = queue.poll();
            if (prePoll.left != null) {
                queue.offer(prePoll.left);
            }
            if (prePoll.right != null) {
                queue.offer(prePoll.right);
            }
            for (int i = 1; i < size; i++) {
                Node poll = queue.poll();
                // 指向该层的下一个节点
                prePoll.next = poll;
                if (poll.left != null) {
                    queue.offer(poll.left);
                }
                if (poll.right != null) {
                    queue.offer(poll.right);
                }
            }
        }
        return root;
    }