二叉树的层序遍历

原题:

给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。

示例:
二叉树:[3,9,20,null,null,15,7],

  1. 3
  2. / \
  3. 9 20
  4. / \
  5. 15 7

返回其层序遍历结果:

[
  [3],
  [9,20],
  [15,7]
]

解题方法:

解法一:

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        if(!root) return {};
        vector<vector<int>> res;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty())
        {
            vector<int> vec;    //注意,在这里声明,让它每层更新
            int size=que.size();//这里是重点,如果直接使用que.size(),要注意它是变化的
            for(int i=0;i<size;i++)
            {
                TreeNode* node=que.front();
                vec.push_back(node->val);
                que.pop();
                if(node->left) que.push(node->left);
                if(node->right) que.push(node->right);
            }
            res.push_back(vec);
        }
        return res;
    }
};

做题小结:

重点即vec的声明需要注意位置,来达到每层更新的目的。

其次则是size变量的使用,来避免que.size()的动态变化。