7-23 还原二叉树

7-23 还原二叉树
分数 25
作者 DS课程组
单位 浙江大学
给定一棵二叉树的先序遍历序列和中序遍历序列，要求计算该二叉树的高度。

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

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

输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
自己的：

#include<stdio.h>
#include<stdlib.h>
typedef struct tree{
    char data;
    struct tree* left;
    struct tree* right;
}tree,*link;
link creat(char pre[],char inorder[],int n);
int get(link root);
int main(){
    int n;
    char pre[55];
    char inorder[55];
    scanf("%d",&n);
    getchar();
    scanf("%s",pre);
    scanf("%s",inorder);
//   	gets(pre);
//   	gets(inorder);

   // puts(pre);
    //puts(inorder);
    link root =( link)malloc(sizeof(tree));
    root=creat(pre,inorder,n);
    int ans=get(root);
    printf("%d",ans);
	 
}
link creat(char pre[],char inorder[],int n){
	if(n==0){
		return NULL;
	}
	link root= (link)malloc(sizeof(tree));

	int index;//根节点在中序遍历中的位置 
	for(int i=0;i<n;i++){
		if(pre[0]==inorder[i]){
			index=i;
			break;
		}
	}
	root->data=pre[0];
	//printf("%c\n",root->data);
	root->left=creat(pre+1,inorder,index);
	root->right=creat(pre+index+1,inorder+index+1,n-index-1); 
	return root;
}
int get(link root){
	if(root==NULL){
		return 0;
	}
	int m=get(root->left);
	int n=get(root->right);
	return m>n?m+1:n+1;
}

抄的：

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
typedef struct tree{
	char data;
	struct tree *left,*right;
}Tree;
//先序第一个确定根节点，中序在根节点左边全是左子树，右边全是右子树 
Tree *creat(char pre[],char mid[],int n)//分别传入先序序列，中序序列，剩余元素的个数
{
	if(n==0)return NULL;
	Tree *t=(Tree*)malloc(sizeof(Tree)); 
	int index=0;
	while(index<n)
	{
		if(mid[index]==pre[0])break;
		index++;
	}
	
	t->data=pre[0];//根节点
	t->left=creat(pre+1,mid,index);//左边确定左子树 
	t->right=creat(pre+index+1,mid+index+1,n-index-1); //右边确定右子树
	return t; 
 } 
int getH(Tree *T)
{
	if(!T)return 0;
	
	int lefth=getH(T->left);
	int righth=getH(T->right);
	
	return lefth>righth?++lefth:++righth;
}
int main()
{	
	int N,HIGH;
	char pre[55],mid[55];
	scanf("%d",&N);
	getchar();
	gets(pre);
	gets(mid);
	Tree *T=creat(pre,mid,N);
	HIGH=getH(T); 
	printf("%d",HIGH);
	return 0;
}