L2-016 愿天下有情人都是失散多年的兄妹 (25 分)
分析:
BFS与set的应用。5代人 相当于BFS 5层
,利用set 的性质 重复元素自动变没, 如果祖先里面有重复元素,那么set在插入前后的大小 不会变化 由此判断是否可以通婚。
代码:
#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
struct node
{
int f,m,sex;//父母编号,性别
};
node people[100000];//p;
int m,b,father,mother;
bool exist[100000];//是否存在在本人Id里 如果不存在就无法知道其父母的编号
int level[100000];//第几代
int main()
{
int n;
char c;
cin>>n;
memset(exist,false,sizeof(exist));
for(int i=0;i<n;i++)
{
scanf("%d %c %d %d",&b,&c,&father,&mother);//编号 性别 父亲 母亲
if(c=='M')
people[b].sex=1;//男
else
people[b].sex=0;//女
people[b].f=father;
people[b].m=mother;
exist[b]=true;//存在本人ID里面 不存在的话等下直接跳过(无法再判断其父母了)
people[father].sex=1;//这里是为了后面直接判断两个人是不是同性的时候需要用到
//即使这两个人没有出现在本人ID里面也可以知道他们的性别
people[mother].sex=0;
}
int k;
cin>>k;
for(int i=0;i<k;i++)
{
fill(level,level+100000,0);//初始化代数
int a,b;
cin>>a>>b;//需要判断的两个人编号
if(people[a].sex==people[b].sex)
{
printf("Never Mind\n");//如果两个人性别一样那么就
continue;//下一轮判断
}
queue<int> q;
q.push(a);
q.push(b);//将两人扔进队列
level[a]=1;
level[b]=1;//是第一代
set<int> s;//用于判断是否有重复元素 如果有 说明有一样的祖先不能通婚
int flag=0;//标记是否为NO了
while(!q.empty())//直到队列不为空
{
int top=q.front();
q.pop();
int size=s.size();
s.insert(top);//放进集合
if(size==s.size())//如果集合数量没有变化的话说明 不能通婚
{
printf("No\n");
flag=1;//标记结束
break;
}
//如果可以继续判断下一个
if(exist[top]==false) continue;//如果他不在本id里面 他父母不知道 所以就不把父母弄进队列了
if(level[top]<=4)//小于四代 就把父母放进去
{
int fa=people[top].f;//父亲
int mo=people[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) printf("Yes\n");
}
return 0;
}