天梯赛L2-016 愿天下有情人都是失散多年的兄妹

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

code：

#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;
} 
	

首先用结构体数组存好各种信息，用level数组标记是第几代，用bfs遍历四代人（level<=4），用set记录出现过的祖先，如果出现了重复的，说明不能通婚。这里有几个需要特别注意的地方：

1. 需要记录父母的性别，这样即使父母没有在输入中作为孩子出现过，也可以进行判断是否是同性进行出五服判断。
2. 判重时，只需要看set的大小有没有变化，如果没有变化说明是重复的。
3. 通过exist数组存放是否在输入中作为孩子出现过，如果没有出现过就不能判断其父母是谁，跳过。