二叉树深度

试题描述

采用用户输入并基于前序遍历元素的方式创建一个存储人名的二叉链树，用户输入两个人的姓名，如果存在，则输出两个人的辈份相差多少，以及是否能够通婚（五代以内（即本人，父母，祖父母，曾祖父母，高祖父母）不可通婚）。

样例输入

ABDFJL…M…G…C.EH…I.KN…OP…

L P

样例输出

1
no

代码

#include <bits/stdc++.h>
using namespace std;
typedef struct node *Tree;
struct node{
    char name;
    Tree left;
    Tree right;
    int Deep;
};
Tree CreateTree(int deep){
    char c;
    Tree tmp;
    cin>>c;
    if(c=='.'){
        tmp=NULL;
    }else{
        tmp=(Tree)malloc(sizeof(struct node));
        tmp->name=c;
        tmp->left=CreateTree(deep+1);
        tmp->right=CreateTree(deep+1);
    }
    if(tmp) tmp->Deep=deep;
    return tmp;
}
int FindNode(Tree T,char c){
    if(!T) return -1;
    if(T->name==c) return T->Deep;
    int a=FindNode(T->left,c);
    int b=FindNode(T->right,c);
    return max(a,b);
}
int main(int argc, char const *argv[])
{
    char a,b;
    Tree T=CreateTree(0);
    cin>>a>>b;
    int deep1=FindNode(T,a);
    int deep2=FindNode(T,b);
    if(deep1!=-1&&deep2!=-1){
        cout<<abs(deep1-deep2)<<endl;
        if(abs(deep1-deep2)<=5)  cout<<"no";
        else  cout<<"yes";
    }
    return 0;
}