C++实现二叉树层序建树及层序遍历

key

  • 按层序遍历顺序读入字符串,'#'表示NULL结点
  • 维持一个先进先出的队列

C++代码实现

cpp 复制代码
#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <queue>
using namespace std;

struct TreeNode {
	char data;
	TreeNode* left;
	TreeNode* right;
};
struct QueueNode {
	TreeNode* parent;//记录父节点
	bool isLeft;//记录父节点左孩子是否插入过
};

void LevelOrder(TreeNode* proot) {
	queue<TreeNode*> pos;
	pos.push(proot);
	while (pos.empty() == false) {
		TreeNode* pCur = pos.front();
		pos.pop();
		printf("%c", pCur->data);
		if (pCur->left != NULL) {
			pos.push(pCur->left);
		}
		if (pCur->right != NULL) {
			pos.push(pCur->right);
		}
	}
}


//可能修改根节点的指向,所以参数必须传根节点指针的引用
void BuildTree(TreeNode* &proot, queue<QueueNode*>& pos, char data) {
	if (data != '#') {
		//申请一个树结点
		TreeNode* pNew = new TreeNode();
		pNew->data = data;

		//申请一个队列结点
		QueueNode* pQueueNode = new QueueNode();
		pQueueNode->parent = pNew;//在队列结点中保存刚创建的新结点位置
		pQueueNode->isLeft = false;//表示新结点的左孩子还没有访问过
		pos.push(pQueueNode);//新结点入队

		if (proot == NULL) {
			proot = pNew;
		}
		else {
			QueueNode* pCur = pos.front();//获取队列队首
			if (pCur->isLeft == false) {//看看队首的左孩子是否插入过
				pCur->parent->left = pNew;
				pCur->isLeft = true;
			}
			else {
				pCur->parent->right = pNew;//左右孩子都插入了,就出队
				pos.pop();
				delete pCur;
			}
		}
	}
	else {//新加入结点是空结点
		QueueNode* pCur = pos.front();
		if (pCur->isLeft == false) {
			pCur->parent->left = NULL;
			pCur->isLeft = true;
		}
		else {
			pCur->parent->right = NULL;
			pos.pop();
			delete pCur;
		}
	}
}

int main() {
	TreeNode* proot = NULL;
	char data;
	queue<QueueNode*> pos;
	while (1) {
		scanf("%c", &data);
		if (data == '\n') {//读到了一行的末尾
			break;
		}
		BuildTree(proot, pos, data);
	}
	LevelOrder(proot);
	return 0;
}
相关推荐
感哥9 小时前
C++ 面向对象
c++
沐怡旸11 小时前
【底层机制】std::shared_ptr解决的痛点?是什么?如何实现?如何正确用?
c++·面试
感哥17 小时前
C++ STL 常用算法
c++
saltymilk1 天前
C++ 模板参数推导问题小记(模板类的模板构造函数)
c++·模板元编程
感哥1 天前
C++ lambda 匿名函数
c++
沐怡旸1 天前
【底层机制】std::unique_ptr 解决的痛点?是什么?如何实现?怎么正确使用?
c++·面试
感哥1 天前
C++ 内存管理
c++
博笙困了2 天前
AcWing学习——双指针算法
c++·算法
感哥2 天前
C++ 指针和引用
c++
感哥2 天前
C++ 多态
c++