天梯赛 L2-030 冰岛人

题意：给你n个人的姓名，他们组成了若干个家族，再给出m个询问，每次询问给出两个人的姓名，问他们能否交往，只要最近公共祖先必须与任何一个人相隔超过5代才能交往。

思路:对每个人的名字进行编号，然后把后缀给去掉，根据名和姓建立若干棵家族树，再用一个节点把这若干棵树给连成一颗树，找最近公共祖先，特判一下0号节点就行了。

代码:

#include<bits/stdc++.h>
using namespace std;
#define inf 0x3f3f3f3f
const int maxn=1e6+9;
struct node{
    int to,next;
}edge[maxn];
int head[maxn],cnt=0,d[maxn],p[maxn][30],flag[maxn],n;
void add(int u,int v){
    edge[cnt].next=head[u];
    edge[cnt].to=v;
    head[u]=cnt++;
}
void dfs(int u){
    for(int i=head[u];i!=-1;i=edge[i].next){
        int to=edge[i].to;
        d[to]=d[u]+1;
        p[to][0]=u;
        dfs(to);
    }
}
void init(){
    for(int j=1;(1<<j)<=maxn;j++){
        for(int i=1;i<=n;i++){
            p[i][j]=p[p[i][j-1]][j-1];
        }
    }
}
bool lca(int a,int b){
    if(d[a]>d[b])swap(a,b);
    int tt=0;
    int f=d[b]-d[a];
    for(int i=0;(1<<i)<=f;i++){
        if((1<<i)&f)b=p[b][i];
    }
    if(a!=b){
        for(int i=(int)log2(maxn);i>=0;i--){
            if(p[a][i]!=p[b][i]){
                a=p[a][i];b=p[b][i];
                tt+=(1<<i);
            }
        }
        tt+=1;
        a=p[a][0];
    }
    
    if(a==0)return true;
    if(tt>=4)return true;
    return false;
}
struct Node{
    string s1,s2;
};
vector<Node>vec;
vector<int>rt;
map<string,int>mp;
int main(){
    memset(head,-1,sizeof(head));
    int i,j,k;
    cin>>n;
    for(i=1;i<=n;i++){
        string s1,s2;
        cin>>s1>>s2;
        vec.push_back({s1,s2});
        mp[s1]=i;
    }
    int len=vec.size();
    for(i=0;i<len;i++){
        string s1,s2;
        s1=vec[i].s1;s2=vec[i].s2;
        if(s2.back()!='n'&&s2.back()!='r'){
            add(0,mp[s1]);
            if(s2.back()=='m')flag[mp[s1]]=1;
        }
        else{
            if(s2.back()=='n'){
                flag[mp[s1]]=1;
                for(j=0;j<4;j++)s2.pop_back();
            }
            else{
                for(j=0;j<7;j++)s2.pop_back();
            }
            add(mp[s2],mp[s1]);
        }
    }
    dfs(0);
    init();
    int m;cin>>m;
    for(i=1;i<=m;i++){
        string s1,s,s2;
        cin>>s1>>s;
        cin>>s2>>s;
        if(!mp[s1]||!mp[s2]){
            cout<<"NA"<<endl;
        }
        else if(flag[mp[s1]]==flag[mp[s2]]){
            cout<<"Whatever"<<endl;
        }
        else{
            if(lca(mp[s1],mp[s2])){
                cout<<"Yes"<<endl;
            }
            else{
                cout<<"No"<<endl;
            }
        }
    }
    //for(int i=1;i<=15;i++)cout<<d[i]<<' ';
    //cout<<d[6]<<endl;
}