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
思路
深搜这两个人5代以内的所有亲属,将他们全部放进set里,若有重复就不行。
坑点:要把该人父母的性别也标注,不标注的话,如果问到这个人的父母,父母是没有性别的。卡了我一天。
代码
#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define PI acos(-1)
#define pb push_back
using namespace std;
typedef pair<int, int> P;
typedef long long ll;
const int N = 3e5 + 19;
const ll mod = 1e9 + 7;
struct node
{
int fa = -1;
int ma = -1;
char gen = 0;
}a[N];
set<int> st;
void insert(int x, int time)
{
if(time == 5 || x == -1)
return ;
st.insert(x);
insert(a[x].fa, time + 1);
insert(a[x].ma, time + 1);
}
bool ok(int y, int time)
{
if(time == 5 || y == -1)
return 0;
if(st.find(y) != st.end())
return 1;
st.insert(y);
if(ok(a[y].fa, time + 1))
return 1;
if(ok(a[y].ma, time + 1))
return 1;
return 0;
}
int main()
{
int T;
cin >> T;
while(T--)
{
int id, fa, ma;
char gen;
cin >> id >> gen >> fa >> ma;
a[id].fa = fa;
a[id].ma = ma;
a[id].gen = gen;
if(fa != -1)
a[fa].gen = 'M';
if(ma != -1)
a[ma].gen = 'F';
}
int k;
cin >> k;
while(k--)
{
int x, y;
cin >> x >> y;
if(a[x].gen == a[y].gen)
{
cout << "Never Mind" << endl;
continue;
}
st.clear();
insert(x, 0);
if(ok(y, 0))
cout << "No" << endl;
else
cout << "Yes" << endl;
}
return 0;
}