1. 程式人生 > 其它 >PTA L2-016 願天下有情人都是失散多年的兄妹 (25 分) (詳解)

PTA L2-016 願天下有情人都是失散多年的兄妹 (25 分) (詳解)

L2-016 願天下有情人都是失散多年的兄妹 (25 分)  

呵呵。大家都知道五服以內不得通婚,即兩個人最近的共同祖先如果在五代以內(即本人、父母、祖父母、曾祖父母、高祖父母)則不可通婚。本題就請你幫助一對有情人判斷一下,他們究竟是否可以成婚?

輸入格式:

輸入第一行給出一個正整數N(2 ≤ N 104),隨後N行,每行按以下格式給出一個人的資訊:

本人ID 性別 父親ID 母親ID
 

其中ID是5位數字,每人不同;性別M代表男性、F代表女性。如果某人的父親或母親已經不可考,則相應的ID位置上標記為-1

接下來給出一個正整數K,隨後K行,每行給出一對有情人的ID,其間以空格分隔。

注意:題目保證兩個人是同輩,每人只有一個性別,並且血緣關係網中沒有亂倫或隔輩成婚的情況。

輸出格式:

對每一對有情人,判斷他們的關係是否可以通婚:如果兩人是同性,輸出Never Mind;如果是異性並且關係出了五服,輸出Yes;如果異性關係未出五服,輸出No

輸入樣例:

24
00001 M 01111 -1
00002 F 02222 03333
00003 M 02222 03333
00004 F 04444 03333
00005 M 04444 05555
00006 F 04444 05555
00007 F 06666 07777
00008 M 06666 07777
00009 M 00001 00002
00010 M 00003 00006
00011 F 00005 00007
00012 F 00008 08888
00013 F 00009 00011
00014 M 00010 09999
00015 M 00010 09999
00016 M 10000 00012
00017 F -1 00012
00018 F 11000 00013
00019 F 11100 00018
00020 F 00015 11110
00021 M 11100 00020
00022 M 00016 -1
00023 M 10012 00017
00024 M 00022 10013
9
00021 00024
00019 00024
00011 00012
00022 00018
00001 00004
00013 00016
00017 00015
00019 00021
00010 00011
 

輸出樣例:

Never Mind
Yes
Never Mind
No
Yes
No
Yes
No
No

大致思路:

把每個人和ta父母的關係記錄下來(樣式像是一棵二叉樹),從本人(第一層)開始按層搜尋,將1-5層出現的人全都記錄下來

對於給定的情侶,先對對一個人搜尋 將第一個人的全部五代人記錄,然後搜尋判斷第二個人 五代所有人當中是否有人在第一個人五代中出現,若出現則No,沒有出現則Yes

主要實現:

定義一個結構體陣列(足夠大),結構體中有fa(父),mo(母)的編號,初值為-1,陣列的下標則代表本人,在輸入的時候將資料記錄

定義一個map<int,char>sex;  用來記錄每個人的性別(相比再開一個超大陣列這樣更省空間),用編號對映性別

定義一個flag標誌,用於記錄第二次深搜時是否出現相同的人,出現則賦1,沒出現則不改變

然後就是搜尋了

 

#include <bits/stdc++.h>
using namespace std;
struct people
{
    int fa = -1;
    int mo = -1;
} peo[100000];      //記錄所有人下標,本人 父 母
map<int, char> sex; //編號對映性別
int flag = 0;       //標誌位,五代中是否出現相同的人
//按層搜尋
void dfs(set<int> &st, int t, int deep)
{
    if (deep == 6)
        return; //搜尋代數1-5,到第6代直接返回
    if (st.count(t) == 1)
    {
        flag = 1;
        return;
    }             //用於第二個人搜尋的判斷,
    st.insert(t); //記錄五代所有人的編號
    if (peo[t].fa != -1)
        dfs(st, peo[t].fa, deep + 1);
    if (peo[t].mo != -1)
        dfs(st, peo[t].mo, deep + 1);
}

void test()
{
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        int t1, t2, t3;
        char c;
        cin >> t1 >> c >> t2 >> t3;
        sex[t1] = c;
        peo[t1].fa = t2;
        peo[t1].mo = t3;
        sex[t2] = 'M'; //注意初始化父母的性別,不用考慮-1的情況,因為編號-1不會在題目中出現
        sex[t3] = 'F';
    }
    int m;
    cin >> m;
    for (int i = 1; i <= m; i++)
    {
        int t1, t2;
        cin >> t1 >> t2; //情侶
        if (sex[t1] == sex[t2])
            cout << "Never Mind" << endl; //性別相同輸出
        else
        {
            set<int> st; //記錄五代所有人
            flag = 0;
            dfs(st, t1, 1); //記錄第一個人的五代
            dfs(st, t2, 1); //記錄並判斷第二個人的五代
            if (flag == 0)
                cout << "Yes" << endl; //標誌位沒變化則五代中沒有相同的人
            else
                cout << "No" << endl;
        }
    }
}

int main()
{
    test();
    return 0;
}

 

有幫助還請多多點贊支援,歡迎轉載!

有任何問題請留言!