基于层序遍历找到二叉树中值为X结点所在层数

算法思想：

算法的主要思路是通过层次遍历的方式逐层访问二叉树的节点。具体步骤如下：

1. 创建一个空队列，并将二叉树的根节点入队。
2. 初始化层数计数器为 0。
3. 当队列非空时，重复以下步骤：
   • 从队列中取出一个节点。
   • 将取出的节点标记为当前层的节点。
   • 如果当前层的节点等于目标节点，则返回当前层数作为目标节点所在的层数。
   • 将当前层的节点的子节点（左子节点和右子节点）依次入队。
   • 完成当前层的节点处理后，层数计数器加一。
4. 如果在遍历完所有节点后仍未找到目标节点，则返回一个表示目标节点不存在的值（如 -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;
}