给定一棵二叉树的先序遍历序列和中序遍历序列,要求计算该二叉树的深度。
输入格式:
输入首先给出正整数N(≤50),为树中结点总数。下面两行先后给出先序和中序遍历序列,均是长度为N的不包含重复英文字母(区别大小写)的字符串。
输出格式:
输出为一个整数,即该二叉树的高度。
输入样例:
9
ABDFGHIEC
FDHGIBEAC
输出样例:
5
#include<iostream>
#include<stdlib.h>
int flag=0;
using namespace std;
typedef struct node
{
struct node *left;
struct node *right;
char data;
}Node;
Node *binarytree(char a[],char b[],int len)
{
int i;
if (len==0)
return NULL;
Node *p;
p=(Node *)malloc(sizeof(Node));
p->data=a[0];
for (i=0;i<len;i++)
{
if (b[i]==a[0])
break;
}
p->left=binarytree(a+1,b,i);
p->right=binarytree(a+i+1,b+i+1,len-1-i);
return p;
}
用一个judge函数来判断他的深度,当深度大于当前找到的最大深度时,更改最大深度。
void judge(Node *p,int level)
{
if (p==NULL)
return ;
if (level>flag)
flag=level;
if (p->left!=NULL)
judge(p->left,level+1);
if (p->right!=NULL)
judge(p->right,level+1);
}
int main()
{
int i,n;
cin>>n;
char pre[n],mid[n];
for (i=0;i<n;i++)
cin>>pre[i];
for (i=0;i<n;i++)
cin>>mid[i];
Node *tree;
tree=(Node *)malloc(sizeof(Node));
tree=binarytree(pre,mid,n);
judge(tree,1);
cout<<flag;
}