LeetCode解题报告 102. Binary Tree Level Order Traversal [easy]

题目描述

Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).

For example:
Given binary tree [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

return its level order traversal as:

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

解题思路

二叉树的层次遍历，输出每一层的节点树，用嵌套的vector表示。

用一个队列来记录当前节点的位置，首先入队根节点root，跟节点出队的同时入队根节点的左右节点；在一次对当前层的遍历中，将左右节点分别当作根节点，出队的同时入队各自对应的左右节点。每次存储的值为当前根节点的val值，一次遍历即存储一行的值。

时间复杂度

bfs，o(n)

代码如下：

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

这里说明一下注释掉的那一行，第一次写的时候没有提前声明l=currlevel.size()，结果导致多出了几次循环，因为在循环中currlevel.size()是变化的，而我们应该使用的是循环之前的值。