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;
}
相关推荐
ulias21233 分钟前
各种背包问题简述
数据结构·c++·算法·动态规划
程序喵大人44 分钟前
分享个C++线程池的实现源码
开发语言·c++·线程池
FL16238631291 小时前
[ubuntu][C++]onnxruntime安装cpu版本后测试代码
linux·c++·ubuntu
要做朋鱼燕2 小时前
【C++】 priority_queue 容器模拟实现解析
开发语言·c++·笔记·职场和发展
励志不掉头发的内向程序员2 小时前
C++进阶——继承 (1)
开发语言·c++·学习
mit6.8244 小时前
并查集|栈
c++
中国胖子风清扬4 小时前
Rust 序列化技术全解析:从基础到实战
开发语言·c++·spring boot·vscode·后端·中间件·rust
j_xxx404_4 小时前
数据结构:栈和队列力扣算法题
c语言·数据结构·算法·leetcode·链表
岁忧5 小时前
(LeetCode 面试经典 150 题) 200. 岛屿数量(深度优先搜索dfs || 广度优先搜索bfs)
java·c++·leetcode·面试·go·深度优先
一枝小雨5 小时前
【OJ】C++ vector类OJ题
数据结构·c++·算法·leetcode·oj题