难度:中等
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
C++:迭代算法
思路:
1、层序遍历可以利用队列来实现
2、根节点先入队,当队列为空时,遍历结束
3、当遍历完一层后,队列中就只剩下一层的节点,可以通过查询队列中的节点数控制一层的迭代数
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<vector<int>> result;
if(!root) return result;
queue<TreeNode*> queue;
queue.push(root); //根节点先入队
while(!queue.empty()){
int count = queue.size();//这一层有多少个节点
vector<int> levelValue;
for(int i = 0; i < count; i++){
TreeNode* p = queue.front();//取出队头
if(p->left != nullptr)
queue.push(p->left);//将队头的左右孩子入队
if(p->right != nullptr)
queue.push(p->right);
levelValue.push_back(p->val);
queue.pop();//弹出队头
}
result.push_back(levelValue);//遍历完一层
}
return result;
}
};