题目链接:https://pintia.cn/problem-sets/994805046380707840/problems/994805061769609216
分析:一开始以为是并查集。。。。。没救了,一看到这种亲属关系的就想往并查集靠
用level数组标记层次,在五层前都可以一遍遍的BFS
用个set,每次从队列取出一个数装进集合,没变化就说明重复过了
注意要给父母也标性别,后面可能出现查询两个没有给出全部信息的人能不能结婚,如果父母没有被单独说过的话,我们就默认父母是合法的,性别也要弄清
exist数组就是来看一个人是不是给出了全部信息,如果不是并且性别相异,我们就认为是可行的
1 #include<bits/stdc++.h> 2 using namespace std; 3 typedef long long ll; 4 const int inf=1<<30; 5 const double pi=acos(-1); 6 const int mod=998244353; 7 const int maxn=1e5+7; 8 const int maxm=6300; 9 struct node{ 10 int f,m,sex; 11 }a[maxn]; 12 bool exist[maxn]; 13 int main(){ 14 int n,id,fa,mo;scanf("%d",&n); 15 char c; 16 for(int i=0;i<n;i++){ 17 scanf("%d %c %d %d",&id,&c,&fa,&mo); 18 exist[id]=true; 19 a[id].f=fa;a[id].m=mo; 20 if(c=='M') a[id].sex=1; 21 if(c=='F') a[id].sex=0; 22 a[fa].sex=1,a[mo].sex=0; 23 } 24 int k,x,y;scanf("%d",&k); 25 for(int i=0;i<k;i++){ 26 scanf("%d%d",&x,&y); 27 if(a[x].sex==a[y].sex){ 28 printf("Never Mind\n"); 29 continue; 30 } 31 int level[maxn]; 32 queue<int> q; 33 q.push(x);q.push(y); 34 level[x]=1,level[y]=1; 35 set<int> s;bool flag=1; 36 while(!q.empty()){ 37 int top=q.front();q.pop(); 38 int size=s.size(); 39 s.insert(top); 40 if(size==s.size()){//插进去一个值却发现集合大小没发生变化,说明另一方曾经插入过 41 printf("No\n"); 42 flag=0;break; 43 } 44 if(!exist[top])continue;//说明他不是我们n个id中的一个,不知道他的父母是谁,不用再考虑了 45 fa=a[top].f,mo=a[top].m; 46 if(fa!=-1&&level[top]<5){ 47 q.push(fa);level[fa]=level[top]+1; 48 } 49 if(mo!=-1&&level[top]<5){ 50 q.push(mo);level[mo]=level[top]+1; 51 } 52 } 53 if(flag) cout<<"Yes\n"; 54 } 55 return 0; 56 }