【链式前向星】实现二叉树的建立和层序遍历

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：
输入第一行给出一个正整数N（≤30），是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。

输出格式：
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例：

4 1 6 3 5 7 2

代码实现

#include<stdio.h>
#include<string.h>
#define MaxSize 150
typedef struct Edge {
	int to, next; //二叉树无需权重
} Edge;
typedef struct { //定义顺序队列
	int data[MaxSize] = {0};
	int front;
	int rear;
} SqQueue;
int head[MaxSize];
int tot = 0;
int i = 0;
int s[MaxSize] = {0};
int N;

//pl为子树在后序序列中的左边界...
void add_edge(Edge edge[], int u, int v) { //u为起点，v为终点,w为权值
	edge[tot].to = v; //tot为空结点指针
	edge[tot].next = head[u];
	head[u] = tot++;//tot为任一个未被使用的空节点指针。
}

void dispose(Edge *root, Edge edge[], int post[], int mid[], int pl, int pr, int ml, int mr) {

	root->to = post[pr];
	int pos = ml;
	int lchild, rchild;
	while (mid[pos] != post[pr]) { //找到子树根在中序中的位置
		pos++;
	}
	int lenL = pos - ml;//左子树的长度
	if (pos < mr) {//由于插入性质，先右子树
		add_edge(edge, root->to, -1);//插入结点编号为root->to的头结点
		rchild = head[root->to];
		dispose(&edge[rchild], edge, post, mid, pl + lenL, pr - 1, pos + 1, mr);
	}

	if (pos > ml) {
		add_edge(edge, root->to, -1);
		lchild = head[root->to];//插入结点编号为root->to的头结点
		dispose(&edge[lchild], edge, post, mid, pl, pl + lenL - 1, ml, pos - 1);
	}

}
int create(Edge edge[]) {
	int post[MaxSize], mid[MaxSize];
	scanf("%d", &N);
	for (int i = 0; i < N; i++)
		scanf("%d", &post[i]);
	for (int i = 0; i < N; i++)
		scanf("%d", &mid[i]);
	dispose(&edge[tot++], edge, post, mid, 0, N - 1, 0, N - 1);//后+中生成树
	return post[N - 1];
}
void levelOrderTraverse(Edge edge[], int Root, SqQueue &Q) { //层序遍历
	if (Q.front != Q.rear) { //若队列非空
		int temp = Q.data[++Q.front];
		s[i++] = temp;
		if (temp != -1) {
			if (head[temp] != -1)//左子树不为空
				Q.data[++Q.rear] = edge[head[temp]].to;
			if (head[temp]!=-1 && edge[head[temp]].next != -1)
				Q.data[++Q.rear] = edge[edge[head[temp]].next].to;
		} else return;
		levelOrderTraverse(edge, Root, Q);
	} else return;
}
int main() {
     Edge edge[MaxSize];//定义链式前向星
	memset(edge, -1, sizeof(Edge)*MaxSize);
	memset(head, -1, sizeof(int)*MaxSize);
	int Root = create(edge);
	SqQueue Q;
	Q.front = Q.rear = -1; //初始化
	Q.data[++Q.rear] = Root;
	levelOrderTraverse(edge, Root, Q);

	for (i = 0; i < N - 1; i++)
		printf("%d ", s[i]);
	printf("%d", s[i]);
	return 0;
}

可改进的点
队列也可用数组实现非链式存储。笔者好懒…快期中了，把前向星搞完就去复习