Leetcode 102. Binary Tree Level Order Traversal(二叉树的层序遍历)

时间:2024-01-05 17:48:32

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]
] 分析:二叉树的层序遍历,使用广度优先搜索,建立一个队列q。
每次将队首temp结点出队列的同时,将temp的左孩子和右孩子入队
用len标记入队后的队列长度
row数组始终为当前层的遍历结果,然后用while(len–)语句保证每一次保存入row的只有一层。
每一次一层遍历完成后,将该row的结果存入二维数组v。同时将row中的元素清空。
 /**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
vector<int> row;
vector<vector<int>> v;
if(root == NULL)
return v;
queue<TreeNode*> node_que;//定义一个队列,元素为节点
TreeNode* temp;//每次都指向队列的第一个元素
node_que.push(root);//将根节点放入队列
while(!node_que.empty()){
int len = node_que.size();//计算当前队列中的元素个数
//每次只下一层的节点放入队列中,且每次只将本层的元素放入row
while(len--){
temp = node_que.front();
node_que.pop();//弹出队列的第一个元素;
row.push_back(temp->val);
if(temp->left != NULL)//保证入队列的不会是空节点
node_que.push(temp->left);
if(temp->right != NULL)//保证入队列的不会是空节点
node_que.push(temp->right);
}
v.push_back(row);
row.clear();
}
return v;
}
};