二叉树的层序遍历-代码随想录

最新推荐文章于 2024-10-06 14:10:45 发布

usa_ham

最新推荐文章于 2024-10-06 14:10:45 发布

阅读量185

点赞数

文章标签：算法数据结构

本文链接：https://blog.csdn.net/usa_ham/article/details/129324525

版权

本文介绍了如何使用层序遍历（广度优先搜索）算法来访问二叉树的节点，分别给出了C++和Java的实现代码。在C++中，利用队列的先进先出特性，存储每层的节点；在Java中，同样采用队列，通过两个循环分别处理队列中的节点和当前层的节点数量。

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

给你一个二叉树，请你返回其按层序遍历得到的节点值。（即逐层地，从左到右访问所有节点）。

思路：1.解题要求要一层一层的遍历二叉树。因此考虑用队列先进先出的特性，来呈现遍历的层次。

2.层序遍历的本质类似于图论中的广度优先遍历，运用过程为我们把每层的孩子节点推进队列，同时按照记录的每层的节点size弹出元素。

C++代码：si

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        queue<TreeNode*> que;
        if (root != NULL) que.push(root);
        vector<vector<int>> result;
        while (!que.empty()) {
            int size = que.size();
            vector<int> vec;
            // 这里一定要使用固定大小size，不要使用que.size()，因为que.size是不断变化的
            for (int i = 0; i < size; i++) {
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if (node->left) que.push(node->left);
                if (node->right) que.push(node->right);
            }
            result.push_back(vec);
        }
        return result;
    }
};

Cpp代码中size为que.size的快照，在while循环中。que.size的大小是不断变化的。

java：

// 102.二叉树的层序遍历
class Solution {
    public List<List<Integer>> resList = new ArrayList<List<Integer>>();

    public List<List<Integer>> levelOrder(TreeNode root) {
        //checkFun01(root,0);
        checkFun02(root);

        return resList;
    }

    //DFS--递归方式
    public void checkFun01(TreeNode node, Integer deep) {
        if (node == null) return;
        deep++;

        if (resList.size() < deep) {
            //当层级增加时，list的Item也增加，利用list的索引值进行层级界定
            List<Integer> item = new ArrayList<Integer>();
            resList.add(item);
        }
        resList.get(deep - 1).add(node.val);

        checkFun01(node.left, deep);
        checkFun01(node.right, deep);
    }

    //BFS--迭代方式--借助队列
    public void checkFun02(TreeNode node) {
        if (node == null) return;
        Queue<TreeNode> que = new LinkedList<TreeNode>();
        que.offer(node);

        while (!que.isEmpty()) {
            List<Integer> itemList = new ArrayList<Integer>();
            int len = que.size();

            while (len > 0) {
                TreeNode tmpNode = que.poll();
                itemList.add(tmpNode.val);

                if (tmpNode.left != null) que.offer(tmpNode.left);
                if (tmpNode.right != null) que.offer(tmpNode.right);
                len--;
            }

            resList.add(itemList);
        }

    }
}