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

题目链接: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 }

猜你喜欢

转载自www.cnblogs.com/qingjiuling/p/10502072.html