二叉树的层次遍历

最新推荐文章于 2024-07-22 22:47:14 发布

子墨777

最新推荐文章于 2024-07-22 22:47:14 发布

阅读量408

点赞数 4

文章标签：算法

本文链接：https://blog.csdn.net/ttomchy/article/details/137237413

版权

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
    vector<vector<int>> result;
    queue<TreeNode*> que;
    if (root == nullptr) {
        return result;
    }
    que.push(root);
    while (!que.empty()) {
        vector<int> tmp;
        int que_size = que.size();
        for (int i = 0; i < que_size; ++i) {
            TreeNode* test = que.front();  // 获取队首节点
            que.pop();  // 弹出队首节点
            tmp.push_back(test->val);  // 将节点值加入当前层的临时向量
            if (test->left) {
                que.push(test->left);  // 将左子节点加入队列
            }
            if (test->right) {
                que.push(test->right);  // 将右子节点加入队列
            }
        }
        result.push_back(tmp);  // 将当前层的节点值向量加入结果向量
    }
    return result;
}

};

二叉树的层次遍历，使用一个队列来进行处理