2020天梯模拟赛L2-4 秀恩爱分得快
L2-4 秀恩爱分得快
古人云：秀恩爱，分得快。
互联网上每天都有大量人发布大量照片，我们通过分析这些照片，可以分析人与人之间的亲密度。如果一张照片上出现了 K 个人，这些人两两间的亲密度就被定义为 1/K。任意两个人如果同时出现在若干张照片里，他们之间的亲密度就是所有这些同框照片对应的亲密度之和。下面给定一批照片，请你分析一对给定的情侣，看看他们分别有没有亲密度更高的异性朋友？
输入格式：
输入在第一行给出 2 个正整数：N（不超过1000，为总人数——简单起见，我们把所有人从 0 到 N-1 编号。为了区分性别，我们用编号前的负号表示女性）和 M（不超过1000，为照片总数）。随后 M 行，每行给出一张照片的信息，格式如下：
K P[1] ... P[K]
其中 K（≤ 500）是该照片中出现的人数，P[1] ~ P[K] 就是这些人的编号。最后一行给出一对异性情侣的编号 A 和 B。同行数字以空格分隔。题目保证每个人只有一个性别，并且不会在同一张照片里出现多次。
输出格式：
首先输出 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
这道题的思路就是用一个char数组来存进每一个输进来的值，若有‘-’，就令sex数组中对应的值为1，表明其性别，之后通过异或运算来表明双方是不是异性，从而判断是否要加亲密度，我们这里采用一个vector数组来存放照片中的每一个人（存入的是对应的绝对值），再写一个函数来表明情侣双方是不是彼此最亲密的人，如果是就输出，如果不是，就去寻找最亲密的人，因为存入的是绝对值，而且for循环本身就是递增顺序，所以不需要再额外排序。
#include<bits/stdc++.h>
using namespace std;
vector<int> v[1010];
bool sex[1010];
int n,m,k,a,b;
void print(int a,int b)//输出函数
{
if(sex[a]) cout<<"-";
cout<<a<<" ";
if(sex[b]) cout<<"-";
cout<<b<<endl;
}
void findmax(int a,int index,double h[])
{
for(int i=0;i<n;i++)
{
if(h[index]==h[i]) print(a,i);
}
}
bool search(int a,int b,double h[],int &index)//index用来表示亲密值最大的值在数组中的位置，并且是引用参数，会因为函数中的改变而改变
{
for(int i=0;i<m;i++)
{
index=v[i].size();
int j;
for(j=0;j<index;j++)
{
if(v[i][j]==a) break;//表明照片中有a
}
if(j>=index) continue;
for(j=0;j<index;j++)
{
if(sex[v[i][j]]^sex[a]) h[v[i][j]]+=1.0/index;//这里通过异或运算来表明是异性
}
}
index=0;
double maxn=-1;
for(int i=0;i<n;i++)
{
if(maxn<h[i])
{
maxn=h[i];
index=i;
}
}
if(h[b]==h[index]) return 1;//如果b恰好是亲密度最大的，则返回1
return 0;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(NULL);//提高cin/cout的效率
cin>>n>>m;
for(int i=0;i<m;i++)
{
cin>>k;
for(int j=0;j<k;j++)
{
char c[6];
cin>>c;
int temp=abs(atoi(c));//将字符串转化为整数再求绝对值
if(c[0]=='-')
sex[temp]=1;//方便后面用异或运算来表明是异性关系（=1表示为女生）
v[i].push_back(temp);//存储每一张照片的人
}
}
char ch[6];
cin>>ch;
a=abs(atoi(ch));
if(ch[0]=='-') sex[a]=1;
cin>>ch;
b=abs(atoi(ch));
if(ch[0]=='-') sex[b]=1;
int ai,bi,isa,isb;//ai,bi表示亲密度最大的值在数组中的位置，isa、isb表示a、b这对情侣是不是对方亲密度最高的人
double ha[1010]={},hb[1010]={};//ha数组存放各个异性与a的亲密度，hb存放各个异性与b的亲密度
isa=search(a,b,ha,ai);
isb=search(b,a,hb,bi);
if(isa&&isb)//如果都是彼此亲密度最大的人，就直接输出
{
print(a,b);
}
else//否则就要找到亲密度最大的人进行输出
{
findmax(a,ai,ha);//先输出A，所以先找A最亲密的异性
findmax(b,bi,hb);//再找B最亲密的人
}
return 0;
}
