剑指 Offer 32 - II. 从上到下打印二叉树 II

最新推荐文章于 2024-09-20 15:28:15 发布

你的航仔

最新推荐文章于 2024-09-20 15:28:15 发布

阅读量96

点赞数 1

分类专栏：树文章标签：二叉树 leetcode c++

本文链接：https://blog.csdn.net/qq_45035709/article/details/108860404

版权

树专栏收录该内容

3 篇文章 0 订阅

订阅专栏

本文介绍了如何使用广度优先搜索(BFS)和深度优先搜索(DFS)来实现从上到下按层打印二叉树。以示例二叉树为基准，展示了两种方法的具体代码实现，每种方法的时间复杂度和空间复杂度均为O(n)。通过层次遍历，可以得到树的层次结构输出。

摘要由CSDN通过智能技术生成

剑指 Offer 32 - II. 从上到下打印二叉树 II
题目：
从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。

例如:
给定二叉树: [3,9,20,null,null,15,7],

返回其层次遍历结果：

[
[3],
[9,20],
[15,7]
]

题解：

BFS

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> ans;
        if(!root)   return ans;     
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
            vector<int>ceng;
            for(int i=q.size();i>0;i--){
                TreeNode*p=q.front();
                q.pop();
                ceng.push_back(p->val);
                if(p->left) q.push(p->left);
                if(p->right) q.push(p->right);
            }
            ans.push_back(ceng);                        
        }
        return ans;
    }
};

时间复杂度O（n），空间复杂度O（n）

2.DFS

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