算法思想:
算法的主要思路是通过层次遍历的方式逐层访问二叉树的节点。具体步骤如下:
- 创建一个空队列,并将二叉树的根节点入队。
- 初始化层数计数器为 0。
- 当队列非空时,重复以下步骤:
- 从队列中取出一个节点。
- 将取出的节点标记为当前层的节点。
- 如果当前层的节点等于目标节点,则返回当前层数作为目标节点所在的层数。
- 将当前层的节点的子节点(左子节点和右子节点)依次入队。
- 完成当前层的节点处理后,层数计数器加一。
- 如果在遍历完所有节点后仍未找到目标节点,则返回一个表示目标节点不存在的值(如 -1)。
这个算法通过使用队列来实现层次遍历的过程,确保每一层的节点按顺序被访问,并在遍历的过程中记录节点所在的层数。当遍历到目标节点时,返回其所在的层数。如果遍历完整个树仍未找到目标节点,则表示目标节点不存在。
int findNodeLevel(bitree root, int target) {
if (root == NULL)
return -1; // 空树,返回 -1 表示节点不存在
Queue q;
initqueue(q); // 初始化队列
enqueue(q, root); // 将根节点入队
int level = 0; // 层数计数器
while (!isempty(q)) {
int size = q.size(); // 当前层的节点数量
level++; // 进入下一层
for (int i = 0; i < size; i++) {
bitree current;
dequeue(q, current); // 出队一个节点
if (current->data == target)
return level; // 找到目标节点,返回当前层数
if (current->lchild != NULL)
enqueue(q, current->lchild); // 将左孩子入队
if (current->rchild != NULL)
enqueue(q, current->rchild); // 将右孩子入队
}
}
return -1; // 遍历完整个树都没有找到目标节点,返回 -1 表示节点不存在
}
完整代码
#include <iostream>
#include <queue>
using namespace std;
// 二叉树节点结构
struct TreeNode {
int data; // 节点数据
struct TreeNode* lchild; // 左子节点指针
struct TreeNode* rchild; // 右子节点指针
};
typedef struct TreeNode* bitree; // 定义二叉树指针类型
typedef queue<bitree> Queue; // 定义队列类型
void initqueue(Queue& q) {
while (!q.empty()) {
q.pop(); // 清空队列
}
}
void enqueue(Queue& q, bitree item) {
q.push(item); // 入队操作
}
void dequeue(Queue& q, bitree& item) {
item = q.front(); // 获取队头节点
q.pop(); // 出队操作
}
bool isempty(Queue& q) {
return q.empty(); // 判断队列是否为空
}
int findNodeLevel(bitree root, int target) {
if (root == NULL)
return -1; // 空树,返回 -1 表示节点不存在
Queue q;
initqueue(q); // 初始化队列
enqueue(q, root); // 将根节点入队
int level = 0; // 层数计数器
while (!isempty(q)) {
int size = q.size(); // 当前层的节点数量
level++; // 进入下一层
for (int i = 0; i < size; i++) {
bitree current;
dequeue(q, current); // 出队一个节点
if (current->data == target)
return level; // 找到目标节点,返回当前层数
if (current->lchild != NULL)
enqueue(q, current->lchild); // 将左孩子入队
if (current->rchild != NULL)
enqueue(q, current->rchild); // 将右孩子入队
}
}
return -1; // 遍历完整个树都没有找到目标节点,返回 -1 表示节点不存在
}
bitree createNode(int data) {
bitree newNode = new TreeNode;
newNode->data = data;
newNode->lchild = NULL;
newNode->rchild = NULL;
return newNode;
}
bitree createBinaryTree() {
bitree root = createNode(1);
root->lchild = createNode(2);
root->rchild = createNode(3);
root->lchild->lchild = createNode(4);
root->lchild->rchild = createNode(5);
root->rchild->lchild = createNode(6);
root->rchild->rchild = createNode(8);
return root;
}
int main() {
bitree root = createBinaryTree();
int target ;
cout << "请输入目标值:";
cin >> target;
int level = findNodeLevel(root, target);
if (level != -1)
cout << "节点 " << target << " 所在的层数为:" << level << endl;
else
cout << "节点 " << target << " 不存在于树中" << endl;
return 0;
}