洛谷P2814 家谱

P2814 家谱

题目背景

现代的人对于本家族血统越来越感兴趣。

题目描述

给出充足的父子关系，请你编写程序找到某个人的最早的祖先。

输入输出格式

输入格式：

输入由多行组成，首先是一系列有关父子关系的描述，其中每一组父子关系中父亲只有一行，儿子可能有若干行，用#name的形式描写一组父子关系中的父亲的名字，用+name的形式描写一组父子关系中的儿子的名字；接下来用?name的形式表示要求该人的最早的祖先；最后用单独的一个$表示文件结束。

输出格式：

按照输入文件的要求顺序，求出每一个要找祖先的人的祖先，格式：本人的名字+一个空格+祖先的名字+回车。

输入输出样例

输入样例#1：

#George

+Rodney

#Arthur

+Gareth

+Walter

#Gareth

+Edward

?Edward

?Walter

?Rodney

?Arthur

$

输出样例#1：

Edward Arthur

Walter Arthur

Rodney George

Arthur Arthur

说明

规定每个人的名字都有且只有6个字符，而且首字母大写，且没有任意两个人的名字相同。最多可能有1000组父子关系，总人数最多可能达到50000人，家谱中的记载不超过30代。

/*

    感觉这题应该是noipT1难度，hash成一个数然后当成一个点记录出度，因为一个人的父亲只可能有一个，所以直接模拟即可，但是最后两个点TLE

*/

#include<iostream>

#include<cstdio>

#include<cstring>

using namespace std;

#define mod 2333333

int cnt,map[],du[],to[];

bool vis[];

char s[][],ch[];

int ha(char c[]){

    int res=;

    for(int i=;i<=;i++)

        res=(res*%mod+(ch[i]-'A'+)*%mod)%mod;

    return res;

}

int main(){

    //freopen("Cola.txt","r",stdin);

    scanf("%s",ch);

    int now,cur;

    while(){

        if(ch[]=='$')break;

        cur=ha(ch);

        if(!vis[cur]){

            cnt++;

            for(int i=;i<;i++)s[cnt][i]=ch[i+];

            map[cur]=cnt;

            vis[cur]=;

        }

        if(ch[]=='#'){

            int fa=cur;

            while(){

                scanf("%s",ch);

                if(ch[]!='+')break;

                int now=ha(ch);

                if(!vis[now]){

                    cnt++;

                    for(int i=;i<;i++)s[cnt][i]=ch[i+];

                    map[now]=cnt;

                    vis[now]=;

                }

                to[now]=fa;

                du[now]++;

            }

        }

        if(ch[]=='?'){

            now=ha(ch);

            //map[now]=cnt;

            cout<<s[map[now]]<<' ';

            while(){

                if(du[now]==)break;

                now=to[now];

            }

            cout<<s[map[now]]<<endl;

            scanf("%s",ch);

        }

    }

}

80分 TLE

#include<iostream>

#include<cstdio>

#include<cstring>

using namespace std;

#define mod 2333333

int cnt,map[],zu[];

bool vis[];

char s[][],ch[];

int ha(char c[]){

    int res=;

    for(int i=;i<=;i++)res=(res*+ch[i]-'A'+)%mod;

    return res;

}

int main(){

    scanf("%s",ch);

    int now,cur;

    for(int i=;i<=;i++)zu[i]=i;

    while(){

        if(ch[]=='$')break;

        cur=ha(ch);

        if(!vis[cur]){

            cnt++;

            for(int i=;i<;i++)s[cnt][i]=ch[i+];

            map[cur]=cnt;

            vis[cur]=;

        }

        if(ch[]=='#'){

            int fa=cur;

            while(){

                scanf("%s",ch);

                if(ch[]!='+')break;

                int now=ha(ch);

                if(!vis[now]){

                    cnt++;

                    for(int i=;i<;i++)s[cnt][i]=ch[i+];

                    map[now]=cnt;

                    vis[now]=;

                }

                zu[map[now]]=zu[map[fa]];

            }

        }

        if(ch[]=='?'){

            now=ha(ch);

            cout<<s[map[now]]<<' ';

            cout<<s[zu[map[now]]]<<endl;

            scanf("%s",ch);

        }

    }

}

另附更改之后WA成50分的代码

#include<bits/stdc++.h>

using namespace std;

map<string,string>father;

string fat(string x){//找爸爸

    if(x!=father[x])

        father[x]=fat(father[x]);

    return father[x];

}

int main(){

    char k;

    string f1,t;

    while(){

        cin>>k;

        if(k=='$') break;

        if(k=='#'){

            cin>>f1;

            if(father[f1]=="") father[f1]=f1;

            continue;

        }

        else if(k=='+'){

            cin>>t;

            father[t]=f1;

        }

        else if(k=='?'){

            cin>>t;

            cout<<t<<' '<<fat(t)<<endl;

        }

    }

    return ;

}

100分 map+并查集