哔哩哔哩2019笔试题--聚会

最新推荐文章于 2022-11-04 10:35:06 发布

转载最新推荐文章于 2022-11-04 10:35:06 发布 · 519 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/programyang/p/11147576.html

文章标签：

#面试

本文通过并查集算法解决了一个社交活动场景下的问题，即计算活动结束后参与者最多能认识多少人。利用并查集的数据结构，有效地处理了n个人之间的m对认识关系，通过不断合并集合来更新参与者之间的连接状态，最终确定了小A能够认识的人数。

小A参加了一个n人的活动，每个人都有一个唯一编号i(i>=0 & i<n)，其中m对相互认识，在活动中两个人可以通过互相都认识的一个人介绍认识。现在问活动结束后，小A最多会认识多少人？

算法：并查集

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int>f;
int find(int x){
    return x==f[x]?x:f[x]=find(f[x]);
}
int main(void){
    int n,idx,m;
    cin>>n>>idx>>m;
    f=vector<int>(n);
    for(int i=0;i<n;i++)f[i]=i;
    int ans(0),b(0);
    while(m--){
        int one, two;
        scanf("%d,%d", &one, &two);
        if(one==idx||two==idx)b++;
        int fx=find(one),fy=find(two);
        if(fx!=fy)f[fx]=fy;
    }
    for(int i=0;i<n;i++)
        if(find(idx)==find(i))ans++;
    cout<<ans-b-1<<endl;
    return 0;
}