按层遍历二叉树

最新推荐文章于 2022-07-09 09:39:15 发布

liuxinglin

最新推荐文章于 2022-07-09 09:39:15 发布

阅读量474

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/liuxinglin/article/details/45277105

版权

LeetCode 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

使用C++中的pair类型来存储树的节点和它的层数

vector<vector<int> > levelOrder(TreeNode *root)
{
    vector<vector<int> > vec;
    if (root == NULL)
        return vec;
    
    queue<pair<TreeNode*, int> > queue;
    pair<TreeNode*, int> pair(root, 0);  //根节点是第0层
    queue.push(pair);
    
    while (!queue.empty()) {
        pair = queue.front();
        queue.pop();
        int val = pair.first->val;
        int level = pair.second;
        
        if (vec.size() <= level)
            vec.push_back(vector<int>());  //增加一个空的新向量存储新一层数据
        vec[level].push_back(val);
        
        
        if (pair.first->left)
            queue.push(make_pair(pair.first->left, level+1));
        if (pair.first->right)
            queue.push(make_pair(pair.first->right, level+1));
    }
    
    return vec;
}