leetcode算法题--从上到下打印二叉树 II

最新推荐文章于 2021-12-07 12:00:55 发布

原创最新推荐文章于 2021-12-07 12:00:55 发布 · 205 阅读

0 ·

CC 4.0 BY-SA版权

Algorithm 专栏收录该内容

375 篇文章

订阅专栏

本文介绍了一种使用队列实现二叉树层次遍历的方法，并提供了完整的C++代码实现。该算法首先将根节点放入队列中，然后依次取出每个节点并将其左右子节点按顺序加入队列，直到队列为空为止。

原题链接：https://leetcode-cn.com/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof/

vector<vector<int>> levelOrder(TreeNode* root) {
    if (root == NULL) return {};
    vector<vector<int>> ans;
    queue<TreeNode*> Q;
    Q.push(root);
    while (!Q.empty()) {
        vector<int> vec;
        for (int i = Q.size(); i > 0 ; i--) {
            TreeNode* t = Q.front();
            Q.pop();
            if (t->left) Q.push(t->left);
            if (t->right) Q.push(t->right); 
            vec.push_back(t->val);              
        }
        ans.push_back(vec);
    }
    return ans;
}