天梯赛L2-30 冰岛人

原题链接

https://pintia.cn/problem-sets/994805046380707840/exam/problems/1111914599412858888?type=7&page=1


题面

(题面较乱,建议还是点击链接去看原题,这里只作预览)
2018年世界杯,冰岛队因1:1平了强大的阿根廷队而一战成名。好事者发现冰岛人的名字后面似乎都有个“松”(son),于是有网友科普如下:
冰岛人沿用的是维京人古老的父系姓制,孩子的姓等于父亲的名加后缀,如果是儿子就加 sson,女儿则加 sdottir。因为冰岛人口较少,为避免近亲繁衍,本地人交往前先用个 App 查一下两人祖宗若干代有无联系。本题就请你实现这个 App 的功能。

输入格式:
输入首先在第一行给出一个正整数 N(1 随后一行给出正整数 M,为查询数量。随后 M 行,每行给出一对人名,格式为:名1 姓1 名2 姓2。注意:这里的姓是不带后缀的。四个字符串均由不超过 20 个小写的英文字母组成。
题目保证不存在两个人是同名的。

输出格式:
对每一个查询,根据结果在一行内显示以下信息:
若两人为异性,且五代以内无公共祖先,则输出 Yes;
若两人为异性,但五代以内(不包括第五代)有公共祖先,则输出 No;
若两人为同性,则输出 Whatever;
若有一人不在名单内,则输出 NA。
所谓“五代以内无公共祖先”是指两人的公共祖先(如果存在的话)必须比任何一方的曾祖父辈分高。

输入样例:
15
chris smithm
adam smithm
bob adamsson
jack chrissson
bill chrissson
mike jacksson
steve billsson
tim mikesson
april mikesdottir
eric stevesson
tracy timsdottir
james ericsson
patrick jacksson
robin patricksson
will robinsson
6
tracy tim james eric
will robin tracy tim
april mike steve bill
bob adam eric steve
tracy tim tracy tim
x man april mikes

输出样例:
Yes
No
No
Whatever
Whatever
NA



解题思路

通过判断姓氏末位的字母来判断性别以及是否是维京人后裔(后裔OR一代),如果是维京人后裔还需要切割尾缀并将父亲姓名存储,查询的时候先判断是不是有不存在的姓名,再判断是不是同性,如果是异性,再去判断是不是五代内有公共祖先。枚举五代内的关系,判断是否有公共祖先~Record[A]用来存储A的性别以及父亲姓名,最后按题意输出

因为发现每个孩子只会对应一个父亲,所以没必要用邻接表建图,直接unordered_map存储关系就行。


代码(CPP)

#include 
using namespace std;
using ll = long long;
using ull = unsigned long long;
using ld = long double;
#define endl '\n';
const int maxn = 1e5 + 10;
const int INF = 0x3fffffff;
const int mod = 1e7 + 9;
struct node
{
    int gender;
    string fa;
};
unordered_map<string, node> rec;
int n;

string check(string a, string b) {
    int level1 = 1;
    string ta = a;
    while (ta != "") {
        int level2 = 1;
        string tb = b;
        while (tb != "") {
            if (ta == tb && (level1 < 5 || level2 < 5)) {
                return "No\n";
            }
            if (level1 >= 5 && level2 >= 5) {
                return "Yes\n";
            }
            tb = rec[tb].fa;
            level2++;
        }
        ta = rec[ta].fa;
        level1++;
    }
    return "Yes\n";
}

void solve() {
    cin >> n;
    for (int i = 1; i <= n; i++) {
        string fname, sname;
        cin >> fname >> sname;
        if (sname.back() == 'n') {
            rec[fname] = {1, sname.substr(0, sname.size() - 4)};
        } else if (sname.back() == 'r') {
            rec[fname] = {0, sname.substr(0, sname.size() - 7)};
        } else if (sname.back() == 'm') {
            rec[fname] = {1, ""};
        } else {
            rec[fname] = {0, ""};
        }
    }

    int k;
    cin >> k;
    while (k--) {
        string fn1, sn1, fn2, sn2;
        cin >> fn1 >> sn1 >> fn2 >> sn2;
        // cout << fn1 << " " << sn1 << " " << fn2 << " " << sn2 << endl;
        if (!rec.count(fn1) || !rec.count(fn2)) {
            cout << "NA\n";
        } else if (rec[fn1].gender == rec[fn2].gender) {
            cout << "Whatever\n";
        } else  {
            cout << check(fn1, fn2);
        }
    }
}

int main() {
    freopen("in.txt", "r", stdin);
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    cout << fixed;
    cout.precision(18);

    int Case = 1;
    // cin >> Case;
    while (Case--)
        solve();
    return 0;
}

你可能感兴趣的:(#,天梯赛,算法,数据结构,团体程序设计天梯赛,GPLT,CCCC,PTA,PAT)