LeCode：（102. 二叉树的层序遍历；107. 二叉树的层序遍历 II）

羊羊羊i2024-01-13 1:33

题目1

题目链接

本题与层序遍历不同的是，是一层一层的输出。

难点：如何一层一层的输出（需要知道每层的个数）

解题思路：
第一层只有一个结点，我们可以使用count计数，记录每层有几个结点，记录第二层有几个结点。然后根据第二层count计数，记录第三层有几个结点。直到遍历完。

cpp 复制代码

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        //根入，遍历，count++
        vector<vector<int>> outPut;  //输出的总数组
        vector<int> t;    //每层的小数组
        queue<TreeNode*> cur;   //队列用来记录层序遍历的结点
        int count = 1; //记录本层个数，第一层为1
        if(root == nullptr)  //树为空，直接返回大数组
        {
            return outPut;
        }
        cur.push(root);   //先将根入队
        while(!cur.empty())  //队列为空，遍历完毕
        {
            int k = count;   //把本层个数给k，
            count = 0;       //count清0， 接着记录下一层
            while(k--)
            {
                TreeNode* tem = cur.front();  
                cur.pop();        
                t.push_back(tem->val);  //将值给小数组
                if(tem->left)
                {
                    count++;
                    cur.push(tem->left);
                } 
                if(tem->right)
                {
                    count++;
                    cur.push(tem->right);
                }
            }
            outPut.push_back(t);   //一层遍历完
            t.clear();  //小数组清空
        }
        return outPut;
    }
};

题目2

题目链接

以为会有新思路，结果官方答案把题目1，最后输出数组翻转了一下。

cpp 复制代码

class Solution {
public:
    vector<vector<int>> levelOrderBottom(TreeNode* root) {

        //根入，遍历，count++
        vector<vector<int>> outPut;  //输出的总数组
        vector<int> t;    //每层的小数组
        queue<TreeNode*> cur;   //队列用来记录层序遍历的结点
        int count = 1; //记录本层个数，第一层为1
        if(root == nullptr)  //树为空，直接返回大数组
        {
            return outPut;
        }
        cur.push(root);   //先将根入队
        while(!cur.empty())  //队列为空，遍历完毕
        {
            int k = count;   //把本层个数给k，
            count = 0;       //count清0， 接着记录下一层
            while(k--)
            {
                TreeNode* tem = cur.front();  
                cur.pop();        
                t.push_back(tem->val);  //将值给小数组
                if(tem->left)
                {
                    count++;
                    cur.push(tem->left);
                } 
                if(tem->right)
                {
                    count++;
                    cur.push(tem->right);
                }
            }
            outPut.push_back(t);   //一层遍历完
            t.clear();  //小数组清空
        }
        reverse(outPut.begin(),outPut.end());
        return outPut;
    }
};