PAT团队赛 L2-016. 愿天下有情人都是失散多年的兄妹

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

输入格式：

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

思路:裸并查集+dfs

#include<iostream>
#include<cstdio>
#include<string>
const int N=1000000; //全局常量
using namespace std;

char sex[N+1];
i
4000
nt fatherid
;
int motherid
;
bool dfs(int xid,int yid,int m){//是否具有五服的关系
if(xid==-1||yid==-1||m>5){
return false;
}
int xf=fatherid[xid];
int xm=motherid[xid];
int yf=fatherid[yid];
int ym=motherid[yid];
if(xf!=-1){
if(xf==yf){
return true;
}
}

if(xm!=-1){
if(xm==ym){
return true;
}
}

if(dfs(xf,yf,m+1)) return true;
if(dfs(xf,ym,m+1)) return true;
if(dfs(xm,yf,m+1)) return true;
if(dfs(xm,ym,m+1)) return true;
return false;
}
void getAnswer(int x,int y){
if(sex[x]==sex[y]){
printf("Never Mind\n");
return ;
}
if(dfs(x,y,2)){//从上一代开始dfs
printf("No\n");
}else{
printf("Yes\n");
}
}
int main(){
int n;
scanf("%d",&n);
//    memset(fatherid,-1,sizeof(fatherid));
//    memset(motherid,-1,sizeof(motherid));
for(int i=0;i<N;i++){
fatherid[i]=-1;
motherid[i]=-1;
}
//    printf("%d\n",n);
for(int i=0;i<n;i++){
int id;
scanf("%d",&id);
scanf(" %c %d %d",&sex[id],&fatherid[id],&motherid[id]);
sex[fatherid[id]]='M';
sex[motherid[id]]='F';
}
int m;
scanf("%d",&m);
for(int i=0;i<m;i++){
int x,y;
scanf("%d %d",&x,&y);
getAnswer(x,y);
}

return 0;
}