剑指offer Leetcode 32-I 二叉树的层序遍历

层序遍历I:单层vector

image-20201207215421746

思想:

​ 用队列控制前后,用size来控制每一层

解法:

class Solution {
    
    
public:
    vector<int> levelOrder(TreeNode* root) {
    
    
        if(root == NULL)
            return {
    
    };
        vector<int>res;
        queue<TreeNode*>my_queue;
        my_queue.push(root);
        while(!my_queue.empty()){
    
    
            int size = my_queue.size();
            //获取每层的个数,只遍历当前层
            while(size--){
    
    
                TreeNode* cur = my_queue.front();
                my_queue.pop();
                res.push_back(cur->val);
                if(cur->left)
                    my_queue.push(cur->left);
                if(cur->right)
                    my_queue.push(cur->right);
            }
        }
        return res;
    }
};

层序遍历II:双层vector

image-20201208095745903

代码:

class Solution {
    
    
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
    
    
        if(root == NULL)
            return {
    
    };
        vector<vector<int>>res;
        queue<TreeNode*>my_queue;
        my_queue.push(root);
        while(!my_queue.empty()){
    
    
            int size = my_queue.size();
            //每层开始时创建一个tmp,结束时将它插入res即可
            vector<int>tmp;
            //获取每层的个数,只遍历当前层
            while(size--){
    
    
                TreeNode* cur = my_queue.front();
                my_queue.pop();
                tmp.push_back(cur->val);
                if(cur->left)
                    my_queue.push(cur->left);
                if(cur->right)
                    my_queue.push(cur->right);
            }
            res.push_back(tmp);
        }
        return res;
    }
};

层序遍历III:之字型双层vector

解法1:迭代器

思想:

​ ●用一个bool型变量reverse来记录是否需要反转,每层reverse取反一次,注意reverse要定义在循环外面

​ ●用vector(tmp.rbegin(), tmp.rend())来获得tmp的倒序,也可以用reverse(tmp.begin(), tmp.end())

代码:

class Solution {
    
    
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
    
    
       if(root == NULL)
            return {
    
    };
        vector<vector<int>>res;
        queue<TreeNode*>my_queue;
        my_queue.push(root);
        //reverse标志位,每一层改变一次,注意要在while的外面定义
        bool reverse = false;
        while(!my_queue.empty()){
    
    
            int size = my_queue.size();
            //每层开始时创建一个tmp,结束时将它插入res即可
            vector<int>tmp;
            //获取每层的个数,只遍历当前层
            while(size--){
    
    
                TreeNode* cur = my_queue.front();
                my_queue.pop();
                tmp.push_back(cur->val);
                if(cur->left)
                    my_queue.push(cur->left);
                if(cur->right)
                    my_queue.push(cur->right);
            }
            if(reverse)
                //tmp.rbegin()和tmp.rend()来倒着插入
                res.push_back(vector<int>(tmp.rbegin(), tmp.rend()));
            else
                res.push_back(tmp);
            reverse = !reverse;
        }
        return res;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_36459662/article/details/113854143