之前学习了,后面忘记了,主要的问题是理解不深、未及时复盘
层序遍历主要通过一个队列实现,对每一层进行处理,每一层使用一个vector保存。在处理每一层时,先计算队列的大小,然后对每一节点进行处理(使用for循环):包括将节点的value保存至vector中,将左、右子树(如果存在)加入至队列中,便于下一次处理。最终返回result。
/**
* 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) {
queue<TreeNode*> que;
vector<vector<int>> result;
if(!root) {
return result;
}
que.push(root);
while(!que.empty()) {
// vec用来保存每一层的节点
vector<int> vec;
int n = que.size();
for(int i = 0; i < n; i++) {
TreeNode* node = que.front();
que.pop();
vec.push_back(node->val);
if(node->left) {
que.push(node->left);
}
if(node->right) {
que.push(node->right);
}
}
// for循环结束,进入下一层
result.push_back(vec);
}
return result;
}
};