1. 题目
给你一个二叉树,请你返回其按层序遍历得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树: [3,9,20,null,null,15,7],
3/ \9 20/ \15 7
返回其层序遍历结果:
[
[3],
[9,20],
[15,7]
]
2. 代码(C++):
/**
* 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) {
TreeNode* cur = root;
deque<TreeNode*> treeList;
vector<vector<int>> result;
// 空根,返回空数组
if (root == nullptr) return result;
// 每一层的节点数
int num = 1;
treeList.push_back(cur);
// 当一层的节点数为 0 时,退出循环
while(num > 0) {
// 记录每一层节点数的临时变量
int num_temp = 0;
// 记录每一层节点对应的值的临时数组
vector<int> temp;
for (int i = 0; i < num; i++) {
cur = treeList.front();
treeList.pop_front();
temp.push_back(cur->val);
if (cur->left != nullptr) {
treeList.push_back(cur->left);
num_temp++;
}
if (cur->right != nullptr) {
treeList.push_back(cur->right);
num_temp++;
}
}
result.push_back(temp);
num = num_temp;
}
return result;
}
};
