L2-016 愿天下有情人都是失散多年的兄妹 (25 分)
呵呵。大家都知道五服以内不得通婚,即两个人最近的共同祖先如果在五代以内(即本人、父母、祖父母、曾祖父母、高祖父母)则不可通婚。本题就请你帮助一对有情人判断一下,他们究竟是否可以成婚?
输入格式:
输入第一行给出一个正整数N(2 ≤ N ≤10 4),随后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
这题考点还是比较明显的,考树
所以vector就少不了了,而且需要注意一个点
如何判断“五服”?以题目输入两个人的编号为DFS的起点,分别进行DFS遍历,如果遍历完了,层数大于或等于5(原本我以为得>5,因为五代以内都不行,但是只大于五得不到样例的输出,只能>=),如果在遍历的过程中,前一个人已经遍历过这个节点,但是这个节点在第二个人的遍历过程中也碰到了,而且小于5,这即是“五服”以内
说白了,就是看看两次DFS过程中在小于5层时,有没有遍历过重复的节点,有就寄
然后就是一些细节上的地方,第一个是vis[]这个数组,每一次循环都需要重置,这里我用了fill,然后就是全局变量flag。
最后一点也非常重要,一开始我也并没有注意到,仔细观察输入样例,输入样例有09999这类>24的编号,但是检查是不是有五服时,没有规定输入的编号必须要<=24,所以每一次输入父母的编号之后都要给他们加上性别,这样就能全面覆盖了!
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 100050;
int n, k;
vector<int> child[maxn];
bool vis[maxn];
string sexList[maxn];
bool flag;
void DFS(int s, int depth)
{
vis[s] = true;
if (depth >= 5)
return;
for (int i = 0; i < child[s].size(); i++)
{
if (vis[child[s][i]] == false)
DFS(child[s][i], depth + 1);
else
flag = false;
}
}
int main()
{
int id, mother, father;
int p1, p2;
char sex;
cin >> n;
for (int i = 0; i < n; i++)
{
cin >> id >> sex >> father >> mother ;
sexList[id] = sex;
if (mother != -1)
{
sexList[mother] = "F";
child[id].push_back(mother);
}
if (father != -1)
{
sexList[father] = "M";
child[id].push_back(father);
}
}
cin >> k;
for (int i = 0; i < k; i++)
{
cin >> p1 >> p2;
if (sexList[p1] == sexList[p2])
cout << "Never Mind" << endl;
else
{
//每次输入之后需重置
fill(vis, vis + maxn, false);
flag = true;
DFS(p1, 1);
DFS(p2, 1);
if (flag == false)
cout << "No"<< endl;
else
cout << "Yes" << endl;
}
}
return 0;
}