PTA L2-016 願天下有情人都是失散多年的兄妹 (25 分) (詳解)
阿新 • • 發佈:2022-03-30
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; }
有幫助還請多多點贊支援,歡迎轉載!
有任何問題請留言!