DS二叉树—二叉树结点的最大距离

题目描述

二叉树两个结点的距离是一个结点经过双亲结点，祖先结点等中间结点到达另一个结点经过的分支数。二叉树结点的最大距离是所有结点间距离的最大值。例如，下图所示二叉树结点最大距离是3，C和D的距离。

二叉树用先序遍历顺序创建，#表示空树。计算二叉树结点最大距离和最大距离的两个结点(假设二叉树中取最大距离的两个结点唯一）。

 

输入

测试次数T

第2行之后的T行，每行为一棵二叉树先序遍历结果（#表示空树）

输出

对每棵二叉树，输出树的结点最大距离和最大距离的结点，输出格式见样例。

 样例输入

3
A##
ABC##EF#G###D##
ABEH###F#K###

样例输出

0:
5:G D
4:H K

代码

#include "bits/stdc++.h"
using namespace std;
const int maxn=1e5+20;
struct NODE{
    char data;
    int d,d2;
    NODE *l,*r,*fa;
    NODE(){
        l=NULL;
        r=NULL;
        fa=NULL;
        d=0;
    }
};
int t,n;
string s;
int maxx;
NODE *qwe=new NODE;
NODE *node=new NODE;

int id=0;
void build(NODE *&head,NODE *fa){
    if(id>=n || s[id]=='#'){
        head=NULL;
        id++;
        return ;
    }
    head=new NODE;
    head->data=s[id];
    head->fa=fa;
    head->d=fa->d+1;
    if(head->d>maxx){
        maxx=head->d;
        qwe=head;
    }
    id++;
    build(head->l,head);
    build(head->r,head);
}

bool check(NODE *a,NODE *b){
    queue<NODE *> qq;
    if(a->l) qq.push(a->l);
    while(!qq.empty()){
        auto now=qq.front();
        qq.pop();
        if(now->data==b->data) return false;
        if(now->l) qq.push(now->l);
        if(now->r) qq.push(now->r);
    }
    return true;
}

void find(NODE *head){
    queue<NODE*> q;
    map<char,int> mp;
    q.push(qwe);
    mp[qwe->data]++;
    qwe->d2=0;
    int ans=0;
    NODE *asd=new NODE;
    while(!q.empty()){
        auto now=q.front();
        q.pop();
        if(now->d2>ans){
            ans=now->d2;
            asd=now;
        }
        if(now->fa!=node && now->fa!=NULL && mp.count(now->fa->data)==0){
            now->fa->d2=now->d2+1;
            q.push(now->fa);
            mp[now->fa->data]++;
        }
        if(now->l!=node &&now->l && mp.count(now->l->data)==0){
            now->l->d2=now->d2+1;
            q.push(now->l);
            mp[now->l->data]++;
        }
        if(now->r!=node &&now->r && mp.count(now->r->data)==0){
            now->r->d2=now->d2+1;
            q.push(now->r);
            mp[now->r->data]++;
        }
    }
    cout<<ans<<":";
    if(ans!=0){
        if(head->l==NULL)
            cout<<asd->data<<" "<<qwe->data;
        else cout<<qwe->data<<" "<<asd->data;
    }
    cout<<endl;
}

int main(){
//    freopen("123.in","r",stdin);
    cin>>t;
    while(t--){
        cin>>s;
        n=s.length();
        id=0;
        NODE *head;
        maxx=0;
        build(head,node);
        find(head);
//        cout<<maxx<<" "<<qwe->data<<endl;
    }
    return 0;
}