L2-016 愿天下有情人都是失散多年的兄妹 (25分) C++

题目描述

呵呵。大家都知道五服以内不得通婚，即两个人最近的共同祖先如果在五代以内（即本人、父母、祖父母、曾祖父母、高祖父母）则不可通婚。本题就请你帮助一对有情人判断一下，他们究竟是否可以成婚？

输入格式

输入第一行给出一个正整数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

---

分析

**此题需要设计一个结构包括个人的性别以及其父母的id号（默认为-1）。
之后每次输入都要把个人及其父母的性别关联起来。
在查找的过程中需要每次把st[]重置，同时对a的祖辈进行dfs（），遍历到的点标记为true.
之后对b同样进行5代之内的dfs，如果此时b的长辈有标记为true的，说明a、b具有共同的祖先，输出“No”，反之则为“Yes”
**

C++ 代码

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int n;
struct people{
	int sex,father=-1,mother=-1;
}peo[N];
bool st[N],flag;
void dfs(int cur,int generation)
{
	if(cur==-1) return;
	if(generation>5) return;
	st[cur]=1;
	dfs(peo[cur].father,generation+1);
	dfs(peo[cur].mother,generation+1);
}
void finddfs(int cur,int generation)
{
	if(cur==-1) return;
	if(generation>5) return;
	if(st[cur]) flag=true;
	finddfs(peo[cur].father,generation+1);
	finddfs(peo[cur].mother,generation+1);
}
int main()
{
	cin>>n;
	int id,father,mother;
	string sex;
	for(int i=0;i<n;i++)
	{
		cin>>id>>sex;
		if(sex=="M") peo[id].sex=1;
		else peo[id].sex=0;
		cin>>peo[id].father>>peo[id].mother;
		peo[peo[id].father].sex=1;
		peo[peo[id].mother].sex=0;	
	}
	cin>>n;
	int a,b;
	for(int i=0;i<n;i++)
	{
		cin>>a>>b;
		if(peo[a].sex==peo[b].sex) puts("Never Mind");
		else{
			memset(st,false,sizeof st);
			dfs(a,1);
			flag=false;
			finddfs(b,1);
			if(flag) puts("No");
			else puts("Yes");
		}
	}
    return 0;
}