题目
大家都知道五服以内不得通婚,即两个人最近的共同祖先如果在五代以内(即本人、父母、祖父母、曾祖父母、高祖父母)则不可通婚。本题就请你帮助一对有情人判断一下,他们究竟是否可以成婚?
输入格式:
输入第一行给出一个正整数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
完整代码如下:
#include<iostream>
#include<set>
#include<queue>
#include<algorithm>
using namespace std;
struct node{
int f,m,sex;
};
node v[100010];
int level[100010];
bool exist[100010];
int main()
{
int n,m,id,father,mother,a,b;
cin>>n;
char c;
for(int i=0;i<n;i++)
{
cin>>id>>c>>father>>mother;
v[id].f=father;v[id].m=mother;
if(c=='M')v[id].sex=0;
else v[id].sex=1;
exist[id]=true;
v[father].sex=0;
v[mother].sex=1;
}
cin>>m;
for(int i=0;i<m;i++)
{
cin>>a>>b;
fill(level,level+100010,0);
if(v[a].sex==v[b].sex)
{
cout<<"Never Mind"<<endl;
continue;
}
queue<int>q;
q.push(a);
q.push(b);
level[a]=1;
level[b]=1;
set<int>s;
int flag = 0;
while(!q.empty())
{
int top = q.front();
q.pop();
int size = s.size();
s.insert(top);
if(size==s.size())
{
cout<<"No"<<endl;
flag = 1;
break;
}
if(exist[top]==false)continue;
if(level[top]<=4)
{
int fa = v[top].f;
int mo = v[top].m;
if(fa!=-1)
{
q.push(fa);
level[fa] = level[top]+1;
}
if(mo!=-1)
{
q.push(mo);
level[mo] = level[top]+1;
}
}
}
if(flag==0)cout<<"Yes"<<endl;
}
return 0;
}
思路分析:
这道题我们采用广度优先搜索,并使用数据结构set集合去减少时间复杂度。具体set如何减少时间复杂度可以看这篇文章: https://blog.csdn.net/yuzhiqiang666/article/details/80628876(建议看完题解后再看)
在代码中node的结构用于存储每一个结点的父母,以node作为内部元素数据类型的数组v[ ]数组level[ ]用于存储这个人属于第几代了,在本题中level[ top ]==1表示的是 top 就是本人,而level[ top ]==2表示top这个人是父母辈的人了,3表示top已经是爷爷奶奶辈的人了,以此类推。exist[top]则表示top这个人是否存在于数组v[ ]当中,如题,如果top这个人不存在于数组v[ ]当中,那么我们就跳过查找top这个人的父母的过程。
struct node{
int f,m,sex;
};
node v[100010];
int level[100010];
bool exist[100010];
本题使用的广度优先遍历非常常见且重要它的一般写法就是首先申请队列(queue<int>q),随后插入队列q的第一个值,但在本题需要插入两个人的值(q.push(a); q.push(b)),也就是把一对新人的值插入。随后将这对新人的父母依次也加入队列,并把对应的儿女(相对于这个父母而言)弹出队列,随后继续将爷爷奶奶加入队列,将父母弹出列队,依次遍历,直到完成对五代内所有亲人的遍历。
在广度遍历过程中比较重要的部分就是如何判断两人是否有共同的五代内亲人,我们采用set集合这个数据结构,解决这个问题,我们首先得到当前集合人的个数,然后再把最近的这个人(也就是top)插入集合后,再次查询当前集合的人数,注意set集合这个数据结构,它的特点是:它所存储的数据是有序(并且是在插入时自动排序)且不重复的,因此如果我们将一个集合中已有的人再一次插入这个集合时,不会导致集合长度的改变。利用此特性,我们可以对比插入前和插入后数组的长度,如果没有发生改变,那么就说明这个人已经在集合里了,也就是说如果他们两个人在一起将会导致五代内存在共同的亲人,此时我们可以直接输出No,并跳出循环。
最后我们还需要判断这个亲人是不是五代以内的人,也就是如果现在的top已经是高祖父母了(即level值为5时),我们不再将它的父母加入队列queue当中继续进行广度遍历。
有任何建议或错误欢迎指正