二叉树的层序遍历

题目描述

给你二叉树的根节点 root ，返回其节点值的 层序遍历 。 （即逐层地，从左到右访问所有节点）。

示例 1：

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

示例 2：
输入：root = [1]
输出：[[1]]

示例 3：
输入：root = []
输出：[]

算法思想

解法一：
定义一个类型为vector< vector < int > > 类型的返回结果集ans,运用递归思想依次进行visit访问，当层数depth大于等于ans.size()时，说明这一层还没来过，这是第一次来，所以得进行扩容,也就是加括号，然后依次遍历左子树和右子树。

解法二：
利用典型的BFS思想进行代码实现，利用队列实现。

代码实现

解法一代码：

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> ans;
        visit(root,0,ans);
        return ans;
    }
    void visit(TreeNode *root, int depth, vector<vector<int>> &ans){
        if(!root) return;
        //depth代表层数，这一层还没来过，这是第一次来，所以要加括号
        if(depth >= ans.size())
            ans.push_back(vector<int>{});
        ans[depth].push_back(root->val);
        visit(root->left, depth + 1, ans);
        visit(root->right, depth + 1, ans);
    }
};

解法二代码：

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
    vector<vector<int>> res;
    if(!root) return res;
    queue<TreeNode*> qu;
    qu.push(root);
    while(!qu.empty())
    {
        vector<int> tmp;
        int len=qu.size();
        for(int i=0;i<len;i++){
            TreeNode* node=qu.front();
            qu.pop();
            tmp.push_back(node->val);
            if(node->left) qu.push(node->left);
            if(node->right) qu.push(node->right);
        }
    res.push_back(tmp);
    }
    return res;
    }
};