LeetCode 102二叉树的层序遍历

LeetCode 102二叉树的层序遍历

• 题目简述：给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

• 输入：[3,9,20,null,null,15,7] 输出：[ [3], [9,20], [15,7] ]

• 宽度优先搜索BFS，每个点进队出队各一次，故时间复杂度为 O(n)。

  使用队列保存每层的所有节点，每次把队列里的原先所有节点进行出队列操作，再把每个元素的非空左右子节点进入队列。因此即可得到每层的遍历。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        if(!root) return res;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty())
        {
            int n = q.size();//记录当前层的元素个数
            vector<int> level;
            //遍历处理当前层的元素同时往队列添加下一层的元素
            for(int i = 0; i < n; i++)
            {
                auto cur = q.front();
                q.pop();
                level.push_back(cur->val);
                if(cur->left) q.push(cur->left);
                if(cur->right) q.push(cur->right);
            }
            if(level.size()) res.push_back(level);
        }
        return res;
    }
};

• 深度优先搜索DFS
  • DFS 不是按照层次遍历的，为了让递归的过程中同一层的节点放到同一个列表中，在递归时要记录每个节点的深度 level。递归到新节点要把该节点放入 level 对应列表的末尾。
  • 当遍历到一个新的深度 level，而最终结果 res 中还没有创建 level 对应的列表时，应该在 res 中新建一个列表用来保存该 level 的所有节点。
  • 递归时要先递归左子树、再递归右子树，对应每一层节点从左到右遍历。

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) 
    {
        vector<vector<int>> res;
        dfs(res, root, 0);
        return res;
    }
    void dfs(vector<vector<int>>& res, TreeNode* root, int level) 
    {
        if (!root) return;
        if (level >= res.size()) 答案层数不够，需要增加一层
            res.push_back(vector<int>());
        res[level].push_back(root->val); //添加到该层末尾
        dfs(res, root->left, level + 1); //去到下一层
        dfs(res, root->right, level + 1);
    }
};