C/C++二叉树的序列化与反序列化-层序遍历--(队列实现)

时间: 2020-06-15 18:46

void serialization(btree* head, queue<int>*& que) {	//层序遍历入队
	if (!head) {	//根指针为空 return
		return;
	}

	queue<btree*> treeQue;	//准备一个队列 (此队列用于层序遍历二叉树)
	treeQue.push(head);	//先将根节点入队

	que->push(head->data);	//再把根节点的数据入队 (此队列用于序列化二叉树)

	while (!treeQue.empty()) {	//栈非空
		btree* p = treeQue.front();
		treeQue.pop();

		if (p->left) {	//节点左子节点非空, 入 treeQue 和 que 队列
			treeQue.push(p->left);
			que->push(p->left->data);
		}
		else {	//若为空则不入 treeQue 队列, 入 que 队列
			que->push(NULL);
		}
		if (p->right) {	//节点右子节点非空, 入 treeQue 和 que 队列
			treeQue.push(p->right);
			que->push(p->right->data);
		}
		else {	//若为空则不入 treeQue 队列, 入 que 队列
			que->push(NULL);
		}
	}
}

btree* buildNode(queue<int>* que) {	//构建节点
	int num = que->front();	//获取队头元素
	que->pop();	//队头元素出队

	if (!num) {	//若出队元素为空, 返回 NULL
		return NULL;
	}

	btree* node = new btree;	//出队元素非空则返回一个节点
	node->data = num;
	return node;
}

btree* bulidTree(queue<int>* que) {	//按层序遍历入队的顺序依次出队, 建树
	if (!que) {	//若 que 为空, 则 return
		return NULL;
	}

	btree* head = buildNode(que);	//建立根节点
	
	queue<btree*> team;	//准备一个队列
	team.push(head);	//先让根节点入队

	while (!team.empty()) {	//队列非空持续循环
		btree* node = team.front();	//获取队头元素
		team.pop();	//队头元素出队

		node->left = buildNode(que);	//对节点建立左子节点	
		node->right = buildNode(que);	//对节点建立右子节点

		if (node->left) {	//左子节点非空, 节点入队
			team.push(node->left);
		}
		if (node->right) {	//右子节点非空, 节点入队
			team.push(node->right);
		}
	}

	return head;
}