L2-016 愿天下有情人都是失散多年的兄妹 (25 分)
呵呵。大家都知道五服以内不得通婚,即两个人最近的共同祖先如果在五代以内(即本人、父母、祖父母、曾祖父母、高祖父母)则不可通婚。本题就请你帮助一对有情人判断一下,他们究竟是否可以成婚?
输入格式:
输入第一行给出一个正整数N(2 ≤ N ≤
1
0
4
10^4
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
题解
从两个id开始,分别做dfs,把再5步之内到的点,都打上标记,如果存在某个点,被标记了两次,就说明不能通婚。
代码
/*
建图,只需要建立指向父母的图
*/
#include <iostream>
#include <vector>
#include <map>
using namespace std;
struct Node
{
int id;
int sex;
int parent[2];
Node()
{
id = 0;
sex = 0;
parent[0]=parent[1]=-1;
}
Node(int _id, char c, int p1, int p2)
{
id = _id, parent[0] = p1, parent[1] = p2;
if(c == 'M') //男
sex = 0;
else //女
sex = 1;
}
};
void dfs(vector<Node> &A, map<int,int> &M, vector<int> &visit, int s, int d, bool &flag)
{
if(d == 1)
return ;
d = d - 1;
for(int i=0;i<2;i++)
{
int pid = A[s].parent[i];
if(pid != -1)
{
int pi = M[pid];
visit[pi]++;
if(visit[pi] > 1)
{
flag =false;
return ;
}
else
{
dfs(A, M, visit, pi, d, flag);
}
}
}
}
bool judge(vector<Node> &A, map<int,int> &M, int &id1, int &id2)
{
int n = A.size();
bool flag = true;
vector<int> visit(n,0);
visit[M[id1]] = 1;
dfs(A, M, visit, M[id1], 5, flag);
visit[M[id2]]++;
if(visit[M[id2]] > 1)
flag = false;
dfs(A, M, visit, M[id2], 5, flag);
return flag;
}
int main()
{
int n;
cin >> n;
map<int, int> M;
char c;
vector<Node> A(n);
for(int i=0;i<n;i++)
{
cin >> A[i].id >> c >> A[i].parent[0] >> A[i].parent[1];
if(c == 'F')
A[i].sex = 1;
M.insert(make_pair(A[i].id, i));
}
int cnt = n;
for(int i=0;i<n;i++)
{
for(int j=0;j<2;j++)
{
int pid = A[i].parent[j];
if(pid != -1)
{
if(M.find(pid) == M.end()) //没有这个人
{
M.insert(make_pair(pid, cnt));
cnt++;
if(j == 0)
A.push_back(Node(pid, 'M', -1, -1));
else
A.push_back(Node(pid, 'F', -1, -1));
}
}
}
}
int k;
cin >> k;
int id1, id2;
Node node1, node2;
for(int i=0;i<k;i++)
{
cin >> id1 >> id2;
node1 = A[M[id1]], node2 = A[M[id2]];
if(node1.sex == node2.sex)
{
cout << "Never Mind" << endl;
}
else
{
if(judge(A, M, node1.id, node2.id))
{
cout << "Yes" << endl;
}
else
{
cout << "No" << endl;
}
}
}
return 0;
}