【PTA 天梯赛】L2-028 秀恩爱分得快（模拟）

古人云：秀恩爱，分得快。

互联网上每天都有大量人发布大量照片，我们通过分析这些照片，可以分析人与人之间的亲密度。如果一张照片上出现了 K 个人，这些人两两间的亲密度就被定义为 1/K。任意两个人如果同时出现在若干张照片里，他们之间的亲密度就是所有这些同框照片对应的亲密度之和。下面给定一批照片，请你分析一对给定的情侣，看看他们分别有没有亲密度更高的异性朋友？

输入格式：

输入在第一行给出 2 个正整数：N（不超过1000，为总人数——简单起见，我们把所有人从 0 到 N-1 编号。为了区分性别，我们用编号前的负号表示女性）和 M（不超过1000，为照片总数）。随后 M 行，每行给出一张照片的信息，格式如下：

K P[1] ... P[K]

其中 K（

输出格式：

首先输出 A PA，其中 PA 是与 A 最亲密的异性。如果 PA 不唯一，则按他们编号的绝对值递增输出；然后类似地输出 B PB。但如果 A 和 B 正是彼此亲密度最高的一对，则只输出他们的编号，无论是否还有其他人并列。

输入样例 1：

10 4
4 -1 2 -3 4
4 2 -3 -5 -6
3 2 4 -5
3 -6 0 2
-3 2

输出样例 1：

-3 2
2 -5
2 -6

输入样例 2：

4 4
4 -1 2 -3 0
2 0 -3
2 2 -3
2 -1 2
-3 2

输出样例 2：

-3 2

#include<bits/stdc++.h>
using namespace std;
int sex[1005],n,m;
int read()
{
    int input=0,sign=0;
    char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')
        ch=getchar();
    if(ch=='-')
    {
        sign=1;       //is girl
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        input=input*10+ch-'0';
        ch=getchar();
    }
    sex[input]=sign;  // girl 1
    return input;
}
int main()
{
    int i,j,k;
    cin>>n>>m;
    vector<vector<int> >photo(n);
    vector<double>pa(n,0.0),pb(n,0.0);
    for(i=0;i<m;i++)
    {
        cin>>k;
        photo[i].resize(k);
        for(j=0;j<k;j++)
            photo[i][j]=read();
    }
    int a=read(),b=read();
    double ma=0,mb=0;
    for(i=0;i<m;i++)
    {
        bool fa=find(photo[i].begin(),photo[i].end(),a)!=photo[i].end();
        bool fb=find(photo[i].begin(),photo[i].end(),b)!=photo[i].end();
        if(fa||fb)// find a or b
        {
            for(j=0;j<photo[i].size();j++)
            {
                if(fa&&sex[a]!=sex[photo[i][j]])     //a存在且对象为异性
                {
                    pa[photo[i][j]]+=1.0/photo[i].size();
                    ma=max(ma,pa[photo[i][j]]);
                }
                else if(fb&&sex[b]!=sex[photo[i][j]])//b存在且对象为异性
                {
                    pb[photo[i][j]]+=1.0/photo[i].size();
                    mb=max(mb,pb[photo[i][j]]);
                }
            }
        }
    }
    if(ma==pa[b]&&mb==pb[a])
        printf("%s%d %s%d\n",sex[a]?"-":"",a,sex[b]?"-":"",b);
    else 
    {
        for(i=0;i<n;i++)
            if(pa[i]==ma)
                printf("%s%d %s%d\n",sex[a]?"-":"",a,sex[i]?"-":"",i);
        for(i=0;i<n;i++)
            if(pb[i]==mb)
                printf("%s%d %s%d\n",sex[b]?"-":"",b,sex[i]?"-":"",i);
    }
    return 0;
}