Codeforces 245G Suggested Friends - 交集set_intersection()

一些人互相是朋友（无向图连一条无向边），现在要添加一个推荐朋友的功能，如果对于x，有y，x与y不是朋友，但是y和x有共同的朋友，并且共同的朋友最多，那么y可以推荐给x。问对于每一个人，有多少人值得推荐。

 

map映射后建图，然后用set_intersection()求交集

这道题学习了一下set_intersection()的用法：

 

set_intersection( fir.begin() , fir.end() , sec.begin() , sec.end() , thr.begin() );

fir、sec、thr都是vector类型，当然也可以是int[]类型

要求fir和sec都是有序的，否则会出现错误。

sort(fir.begin(),fir.end());

 

该函数返回值为交集的尾指针，就是类似于thr.end()的概念。

 

这道题就对每个人求其他人的交集，计算出交集中有多少人即可。

 

#include<bits/stdc++.h>

#define eps 1e-9

#define ALL(x) x.begin(),x.end()

#define INS(x) inserter(x,x.begin())

#define FOR(i,j,k) for(int i=j;i<=k;i++)

#define MAXN 1005

#define MAXM 40005

#define INF 0x3fffffff

using namespace std;

typedef long long LL;

int i,j,k,n,m,x,y,T,ans,big,cas,num,len,an,bn,ans_num,cur;

bool flag;

map <string,int> mp;

vector <int> st[10002];

int vis[10002];

int tmp[10002];

string a,b,rev[10002];

int main()

{

    scanf("%d",&n);

    

    for (i=1;i<=n;i++) st[i].clear();

    mp.clear();

    memset(tmp,0,sizeof(tmp));

    num=0;

    an=bn=0;

    ans_num=0;

    cur=0;

    ans=0;

    

    for (i=1;i<=n;i++)

    {

        cin>>a>>b;

        if (!mp.count(a))

        {

            mp[a]=++num;

            rev[num]=a;

            an=num;

        }else an=mp[a];

        

        if (!mp.count(b))

        {

            mp[b]=++num;

            rev[num]=b;

            bn=num;

        }else bn=mp[b];

        

        st[an].push_back(bn);

        st[bn].push_back(an);

    }

    

    for (i=1;i<=num;i++)

    {

        sort(st[i].begin(),st[i].end()); 

    } 

    

    

    printf("%d\n",num);

    for (i=1;i<=num;i++)

    {

        int pre=0;

        

        cur=0;ans=0;ans_num=0;

        

        memset(vis,0,sizeof(vis));

        for (vector <int> ::iterator it=st[i].begin();it!=st[i].end();it++)

        {

            vis[*it]=1;

        }

        vis[i]=1;

        

        

        for (j=1;j<=num;j++)

        {

            if (vis[j]) continue;

            

            cur=set_intersection(st[i].begin(),st[i].end(),st[j].begin(),st[j].end(),tmp)-tmp;

            if (cur>ans_num)

            {

                ans_num=cur;

                ans=1;

            }else

            if (cur==ans_num)

            {

                ans++;

            }

                

        }

        

        cout<<rev[i]<<" "<<ans<<endl;

    }

     

    return 0;

}