数据结构课设 还原二叉树

1、

5-9 还原二叉树   (25分)

给定一棵二叉树的先序遍历序列和中序遍历序列，要求计算该二叉树的高度。

输入格式:

输入首先给出正整数N（$\le$50），为树中结点总数。下面两行先后给出先序和中序遍历序列，均是长度为N的不包含重复英文字母（区别大小写）的字符串。

输出格式:

输出为一个整数，即该二叉树的高度。

输入样例:

9
ABDFGHIEC
FDHGIBEAC

输出样例:

5

不建树：

#include <iostream>
#include <cstring>
using namespace std;
int getHeight(string pro, string in) {
	char c = pro.at(0);
	int i = in.find(c);
	int len = pro.size();
	if(len == 1) return 1;
	int leftHeight ,rightHeight;
	leftHeight = rightHeight = 0;
	if(i >= 1) {
		leftHeight = getHeight(pro.substr(1, i), in.substr(0, i - 1));
	}
	if(i < len - 1) {
		rightHeight = getHeight(pro.substr(i + 1, len - 1), in.substr(i + 1, len - 1));
	}
	return 1 + max(leftHeight, rightHeight);
}
int main() {
    string pro, in;
    int n;
    cin >> n;
    cin >> pro >> in;
    cout << getHeight(pro, in);
    return 0;
}

2、

L2-006. 树的遍历

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出一个正整数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

题目链接：https://www.patest.cn/contests/gplt/L2-006

建树

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
struct node {
	int num;
	node * left, *right;
};
node * getpre(int *pos, int *in, int n) {
	if(n == 0) return 0;
	int i;
	for(i = 0; in[i] != pos[n - 1]; i++);
	node *rt = new node;
	rt->num = pos[n - 1];
	rt->left = getpre(pos, in, i);
	rt->right = getpre(pos + i, in + i + 1, n - i - 1);
	return rt;
}
int main() {
	int n;
    scanf("%d", &n);
    int pos[40], in[40];
    int i;
    for(i = 0; i < n; i++) {
    	scanf("%d", pos + i);
	}
	for(i = 0; i < n; i++) {
		scanf("%d", in + i);
	}
	node * rt = getpre(pos, in, n);
	queue<node*> q;
	q.push(rt);
	while(!q.empty()) {
		node * t = q.front();
		q.pop();
		printf("%d", t->num);
		if(--n) printf(" ");
		else printf("\n");
		if(t->left) q.push(t->left);
		if(t->right) q.push(t->right);
	}
    return 0;
}