题解 | 求二叉树的层序遍历

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <queue>
#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型vector<vector<>>
     */
    vector<vector<int> > levelOrder(TreeNode* root) {
       queue<TreeNode*> q1,q2;
       vector<vector<int>> res = {};

       if(root) q2.push(root);
       while(!q2.empty()){
            vector<int> temp;
            q1 = q2;
            queue<TreeNode*>().swap(q2);
            while(!q1.empty())
            {
                temp.push_back(q1.front()->val);
                if(q1.front()->left)q2.push(q1.front()->left);
                if(q1.front()->right)q2.push(q1.front()->right);
                q1.pop();
            }
            res.push_back(temp);
       }
       return res;
    }
};