用c++编写程序对二叉树进行先序遍历,并打印层号
时间: 2024-03-05 21:51:58 浏览: 69
C++数据结构与算法二叉树的层序遍历代码及注释
5星 · 资源好评率100%
以下是用 C++ 编写程序对二叉树进行先序遍历,并打印层号的代码示例:
```cpp
#include <iostream>
#include <queue>
using namespace std;
// 二叉树结点定义
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
// 二叉树先序遍历函数,同时打印结点所在的层号
void preorderTraversal(TreeNode* root) {
if (root == NULL) {
return;
}
queue<pair<TreeNode*, int>> q;
q.push(make_pair(root, 0));
while (!q.empty()) {
TreeNode* node = q.front().first;
int level = q.front().second;
q.pop();
cout << "层号:" << level << ",结点值:" << node->val << endl;
if (node->left != NULL) {
q.push(make_pair(node->left, level + 1));
}
if (node->right != NULL) {
q.push(make_pair(node->right, level + 1));
}
}
}
int main() {
// 构造一个二叉树
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->left = new TreeNode(4);
root->left->right = new TreeNode(5);
root->right->left = new TreeNode(6);
root->right->right = new TreeNode(7);
// 先序遍历二叉树,并打印层号
preorderTraversal(root);
return 0;
}
```
该程序中,我们使用了一个队列来辅助进行层次遍历,每次从队列中取出一个结点进行处理,同时将其左右子结点加入队列中,并且将其层号加一。每次处理一个结点时,我们都打印其所在的层号以及结点的值。这样就能够实现对二叉树的先序遍历,并打印层号了。
阅读全文