L3-3. 社交集群-CFANZ编程社区

L3-3. 社交集群

时间限制

1000 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入格式：

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

K_i: h_i[1] h_i[2] ... h_i[K_i]

其中K_i（>0）是第i个人的兴趣的数量，h_i[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出格式：

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

输入样例：

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

题解：vector数组为活动号，保存选择该活动的人员编号。
从第一个人i开始查找，所有和本人在同一活动中的人全部入队，并标记上i，并对队的每个人进行相同查找，直到队空。
i++，在进行下一个没有查找过的人的查找到结束。
最后计算标记数组中共出现的标号和数量，排序输出。

AC代码：

#include<iostream>
#include<stdio.h>
#include<math.h>
#include<string.h>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
vector <int >a[1010];
int main(){
  int N;
  scanf("%d",&N);
  int mmax = 0;
  for(int i=1;i<=N;i++){
    int n;
    scanf("%d:",&n);
    for(int j=0;j<n;j++){
      int num;
      scanf("%d",&num);
      if(mmax<num){
        mmax = num;
      }
      a[num].push_back(i); 
    }
  }
  int x[1010],y[1010];        //x:person   y:problem
  memset(x,0,sizeof(x));
  memset(y,0,sizeof(y));
  for(int i=1;i<=N;i++){//ren
    if(!x[i]){
      x[i] = i;
      queue<int>q;
      q.push(i);
      while(!q.empty()){
        int p = q.front();
        q.pop();
        for(int j=0;j<=mmax;j++){
          if(a[j].size()!=0&&!y[j]&&find(a[j].begin(),a[j].end(),p)!=a[j].end()){
            y[j] = 1;
            for(int k=0;k<a[j].size();k++){
              if(!x[a[j][k]]){
                x[a[j][k]] = i;
                q.push(a[j][k]);
              }
            }
            
          }
        }
      }
    }
  }
  int c[1100];
  memset(c,0,sizeof(c));
  for(int i=1;i<=N;i++){
    c[x[i]]++;
  }
  sort(c,c+N+10,greater<int>());
  int time = 0;
  for(int i=0;c[i];i++){
    time++;
  }
  printf("%d\n",time);
  if(time){
    printf("%d",c[0]);
  for(int i=1;c[i];i++){
    printf(" %d",c[i]);
  }
  printf("\n");
  }
  
  return 0;
}