c++ 给你二叉树的根节点 root ,返回其节点值的 层序遍历
时间: 2023-10-24 10:30:18 浏览: 131
可以使用BFS(广度优先搜索)实现二叉树的层序遍历。具体实现步骤如下:
1. 创建一个队列,并将二叉树的根节点入队。
2. 当队列不为空时,循环执行以下操作:
- 弹出队首元素,并将该元素的值存储到结果数组中。
- 如果该元素的左子节点不为空,将其左子节点入队。
- 如果该元素的右子节点不为空,将其右子节点入队。
下面是对应的 C++ 代码实现:
```c++
#include <iostream>
#include <vector>
#include <queue>
using namespace std;
// Definition for a binary tree node.
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
vector<int> levelOrder(TreeNode* root) {
vector<int> res;
if (root == nullptr) {
return res;
}
queue<TreeNode*> q;
q.push(root);
while (!q.empty()) {
int n = q.size();
for (int i = 0; i < n; i++) {
TreeNode* node = q.front();
q.pop();
res.push_back(node->val);
if (node->left != nullptr) {
q.push(node->left);
}
if (node->right != nullptr) {
q.push(node->right);
}
}
}
return res;
}
int main() {
TreeNode* root = new TreeNode(3);
root->left = new TreeNode(9);
root->right = new TreeNode(20);
root->right->left = new TreeNode(15);
root->right->right = new TreeNode(7);
vector<int> res = levelOrder(root);
for (int i = 0; i < res.size(); i++) {
cout << res[i] << " ";
}
cout << endl;
return 0;
}
```
输出结果为:3 9 20 15 7。
阅读全文