L3-1 社交集群(30 分)经典算法之并查集

L3-1 社交集群(30 分)
当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式:
输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:

这里写图片描述
输出格式:
首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔,行末不得有多余空格。

输入样例:
8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4
输出样例:
3
4 3 1

在这里面谈谈对sort排序的使用,sort()函数一般有三个参数,平时我们会在main函数上面定义一个排序规则,如果想默认使用降序排列的话;以下给出两种算法来实现排序功能:
代码一:

sort(num, num + 1010, greater<int>());

代码二:

bool cmp(int a,int b){
    return a>b;
}
sort(arr2,arr2+1002,cmp);
#include<bits/stdc++.h>
using namespace std;
int f[1002];
int fun(int x){
    if(f[x]==x){
        return f[x];
    }
    f[x]=fun(f[x]);
    return f[x];
}
void group(int a,int b){
    int fa=fun(a);
    int fb=fun(b);
    if(fa!=fb){
        f[fa]=fb;
    }
    return;
}
bool cmp(int a,int b){
    return a>b;
}
int main(){
    int a,b,c,t;
    int arr[1002]={0};
    cin>>t;
    for(int i=0;i<1002;i++){
        f[i]=i;
    }
    for(int i=0;i<t;i++){
        scanf("%d:",&a);
        scanf("%d",&b);
        arr[i]=b;//数组保存了每个编号所在的集合
        for(int j=1;j<a;j++){
            cin>>c;
            group(b,c);
        }
    }
    int arr2[1002];
    memset(arr2,0,sizeof(arr2));
    int ant=0;
    for(int i=0;i<t;i++){
       int q=fun(arr[i]);
       arr2[q]+=1;
    }
    for(int i=0;i<1002;i++){
        if(arr2[i]!=0){
            ant++;
        }
    }
    cout<<ant<<endl;
    sort(arr2,arr2+1002,cmp);
    //sort(num, num + 1010, greater<int>());
    for(int j=0;j<ant-1;j++){
        printf("%d ",arr2[j]);
    }
    printf("%d",arr2[ant-1]);
    return 0;
}


猜你喜欢

转载自blog.csdn.net/weixin_38505045/article/details/80469867
今日推荐