【HBU】4-14 还原二叉树 (15分)

傷城~ 2022-12-20 13:58 28阅读 0赞

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

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

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

**输入样例:**

9
    ABDFGHIEC
    FDHGIBEAC

**输出样例:**

## 两个知识点，根据前序、中序遍历还原二叉树；求二叉树的高度。 ##

可以参看视频讲解 [浙江大学-数据结构][-]

#include <iostream>
    using namespace std;
    char pre[52], in[52];
    typedef struct bitnode{
        
    	char data;
    	struct bitnode *lchild, *rchild;
    }BiTNode, *BiTree;
    BiTree buildTree(int root, int start, int end){
        
    	if(start > end) return NULL;
    	BiTree p = (BiTNode *)malloc(sizeof(BiTNode));
    	p->data = pre[root];
    	int i, cnt = 0;
    	for(i = start; i < end; i++){
        
    		cnt++;
    		if(in[i] == pre[root])
    			break;
    	}
    	p->lchild = buildTree(root+1, start, i-1);
    	p->rchild = buildTree(root+cnt, i+1, end);
    	return p;
    }
    int getHeight(BiTree bt){
        
    	if(bt){
        
    		int l, r;
    		l = getHeight(bt->lchild);
    		r = getHeight(bt->rchild);
    		return max(l, r)+1;
    	}
    	return 0;
    }
    int main(){
        
    	int n;
    	cin >> n;
    	cin >> pre >> in;
    	BiTree p = buildTree(0, 0, n-1);
    	int hight = getHeight(p);
    	cout << hight;
    }

## 附：由后序、中序遍历还原二叉树代码 ##

BiTree setTree(int root, int begin, int end, int post[], int in[]){
        
    	if(begin > end) return NULL;
    	BiTree t = (BiTNode*)malloc(sizeof(BiTNode));
    	t->data = post[root];
    	int i, cnt = 0;
    	for(i = begin; i < end; i++){
        
    		cnt++;
    		if(in[i] == post[root])
    			break;
    	}
    	t->lchild = setTree(root-1-(end-i), begin, i-1, post, in);
    	t->rchild = setTree(root-1, i+1, end, post, in);
    	return t;
    }

[-]: https://www.bilibili.com/video/BV1JW411i731?p=37