L3-003. 社交集群

题目

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

输入格式：

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

Ki: hi[1] hi[2] … hi[Ki]

其中Ki（>0）是第i个人的兴趣的数量，hi[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

思路

• 兴趣做节点，兴趣下挂人
• 同一个人的兴趣相连通
• 搜索兴趣图的连通块（即集群），将人放入set去重，统计块内人数
• 排序输出每个集群的人数

代码

#include<iostream>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;

vector<int> hi[1005];
bool f[1005];
vector<int> edge[1005];
set<int,less<int> > set1;
vector<int> v;

void dfs(int v){
  f[v]=0;
  int p;
  for(int i=0;i<hi[v].size();i++){
    p=hi[v][i];
    set1.insert(p);
  }
  for(int i=0;i<edge[v].size();i++){
    int w=edge[v][i];
    if(f[w]){
      dfs(w);
    }
  }
}
int main(){
  int n;
  int m;
  int a,b;
  cin>>n;
  for(int i=1;i<=n;i++){
    cin>>m;
    getchar();
    a=0;
    for(int j=0;j<m;j++){
      cin>>b;
      f[b]=1;
      hi[b].push_back(i);
      if(a!=0){

        edge[a].push_back(b);
        edge[b].push_back(a);
      }
      a=b;
    }

  }

  for(int i=1;i<=1000;i++){
    if(f[i]){

      set1.clear();
      dfs(i);//dfs(b)是咋写出来的～～～ 
      int num=set1.size();
      v.push_back(num);
    }
  }
  cout<<v.size()<<endl;
  sort(v.begin(),v.end()); 
  for(int i=v.size()-1;i>0;i--){
    cout<<v[i]<<" ";
  }
  cout<<v[0]<<endl;
  return 0;
}