代码随想录算法训练营|102.二叉树的层序遍历① 107.二叉树的层序遍历②

102.二叉树的层序遍历①【中等】

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

要想实现二叉树的层序遍历，需要队列这个数据结构。逐层将二叉树的节点加入队列，并弹出到结果数组（result）中，直到遍历结束。

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

由于题目要求的输出格式是动态数组的嵌套，所以定义了两个 vector<int> (result以及vec）。需要注意的是，别忘了将vec push_back into result...（要被自己乐死了，第一次写的时候落下了这句，结果输出的全是空，想了半天才发现 ）。

107.二叉树的层序遍历②【中等】

给你二叉树的根节点  root  ，返回其节点值  自底向上的层序遍历  。 （即按从叶子节点所在层到根节点所在的层，逐层从左向右遍历）。

和102题相比只是输出的数组顺序颠倒一下而已，直接采用暴力的reverse函数好了...

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {
        queue<TreeNode*> que;
        vector<vector<int>> result;
        if(root!= NULL)que.push(root);
        while(!que.empty()){
            int size = que.size();
            vector<int> vec;
            for(int i=0;i<size;i++){
                TreeNode* node = que.front();
                que.pop();
                vec.push_back(node->val);
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            result.push_back(vec);
        }
        reverse(result.begin(),result.end());
        return result;
    }
};

以上就是全部内容啦，别忘了一键三连，谢谢喵 

         ^    ^

~o( =∩ω∩= )o~