栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > C/C++/C#

102. Binary Tree Level Order Traversal

C/C++/C# 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

102. Binary Tree Level Order Traversal

题意

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

题解

很明显,要采用层序遍历。采用队列存储数据。把每一层的节点都添加到同一个数组中即可,问题的关键在于遍历同一层节点前,必须事先算出同一层的节点个数有多少(即队列已有元素个数),因为 BFS 用的是队列来实现的,遍历过程中会不断把左右子节点入队。

代码
class Solution {
public:
    vector> levelOrder(TreeNode* root) {
        //层序遍历
        
        if(root==NULL) return {};
        queue q;
        q.push(root);
        vector> res;
        while(!q.empty()){
            int size = q.size();
            vector level;
            for(int i=0; ival);
                if(temp -> left !=NULL) q.push(temp->left);
                if(temp -> right != NULL) q.push(temp -> right);                
            }
            res.push_back(level);  
        }
        return res;
        
    }
};

DFS:

         DFS 可以用递归来实现,其实只要在递归函数上加上一个「层」的变量即可,只要节点属于这一层,则把这个节点放入相当层的数组里,代码如下:

class Solution {
public:
    vector> levelOrder(TreeNode* root) {
        vector>res;
        DFS(res, root, 0);
        return res;
    }
    
    void DFS(vector>& res, TreeNode* root, int level){
        if(!root) return;
        if(level == res.size()) res.push_back(vector());
        res[level].push_back(root->val);
        DFS(res, root->left, level + 1);
        DFS(res, root->right, level + 1);
    }
};

 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/916916.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号