原始代码
void maxcodenumber(bitree *t)
{
if(t!=NULL){
Queue q;
initQueque(q);
enqueue(q,t);
int maxcodenumber=1;
while (isempty(q))
{
int size=q.size();
if(size>maxcodenumber)
maxcodenumber=size;
enqueue(q, t->lchild); // 将左孩子入队
enqueue(q, t->rchild); // 将右孩子入队
}
}
cout<<maxcodenumber<<endl;
}
修改后代码
#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(); // 出队操作
}
void maxcodenumber(bitree t) {
if (t != NULL) {
Queue q;
initQueue(q);
enqueue(q, t);
int maxCodeNumber = 0;
while (!q.empty()) {
int size = q.size();
if (size > maxCodeNumber) {
maxCodeNumber = size;
}
for (int i = 0; i < size; i++) {
bitree current;
dequeue(q, current);
if (current->lchild != NULL) {
enqueue(q, current->lchild); // 将左孩子入队
}
if (current->rchild != NULL) {
enqueue(q, current->rchild); // 将右孩子入队
}
}
}
cout << "二叉树各层节点数值中最大值: " << maxCodeNumber << endl;
}
}
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(7);
return root;
}
int main() {
bitree root = createBinaryTree();
maxcodenumber(root);
return 0;
}
注意:
1.在while里出队操作的位置,因为是要对层中的每个节点做处理,因此要加一个for循环,再循环里面出队,这点和根据层序遍历求x的层数那道题的想法一样
附上代码
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 表示节点不存在
都是在for循环里做处理