题目描述
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()是变化的,而我们应该使用的是循环之前的值。