Binary Tree Level Order Traversal

最新推荐文章于 2024-10-15 19:28:18 发布

Dr-rong

最新推荐文章于 2024-10-15 19:28:18 发布

阅读量253

点赞数

分类专栏： LeetCode 文章标签： algorithm leetcode

本文链接：https://blog.csdn.net/durong123123123/article/details/43112541

版权

LeetCode 专栏收录该内容

137 篇文章 0 订阅

订阅专栏

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int> > res;
    vector<vector<int> > levelOrder(TreeNode *root) {
        queue<TreeNode*> mq1;
        queue<TreeNode*> mq2;
        if(root==NULL)
        {
            return res;
        }
        mq1.push(root);
        while((!mq1.empty())||(!mq2.empty()))
        {
            if(mq1.empty())
            {
                //pop mq2 and fill mq1
                vector<int> tmp;
                while(!mq2.empty())
                {
                    TreeNode* nn=mq2.front();
                    mq2.pop();
                    tmp.push_back(nn->val);
                    if(nn->left!=NULL)
                    {
                        mq1.push(nn->left);
                    }
                    if(nn->right!=NULL)
                    {
                        mq1.push(nn->right);
                    }
                }
                res.push_back(tmp);
            }
            else
            {
                //pop mq1 and fill mq2
                vector<int> tmp;
                while(!mq1.empty())
                {
                    TreeNode* nn=mq1.front();
                    mq1.pop();
                    tmp.push_back(nn->val);
                    if(nn->left!=NULL)
                    {
                        mq2.push(nn->left);
                    }
                    if(nn->right!=NULL)
                    {
                        mq2.push(nn->right);
                    }
                }
                res.push_back(tmp);
            }
        }
        return res;
    }
};