古人云:秀恩爱,分得快。
互联网上每天都有大量人发布大量照片,我们通过分析这些照片,可以分析人与人之间的亲密度。如果一张照片上出现了 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
思路:
注意0和-0的处理,然后输入每张照片的时候,分别存到F[i],和M[i]里,表示第i张照片里女生有谁,第i张照片里男生有谁。
然后再分别查没长照片里有没有a,b,然后计算出其和别人对应的亲密度,并存到map里
(之前没写出来,今天竟然写出来啦,哈哈哈)
代码如下:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<queue>
#include<stack>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
typedef pair<int,int>P;
const int INF=0x3f3f3f3f;
const int N=1015,mod=32767;
int t[N];
char s[10],s1[10];
set<int>M[N],F[N];
map<int,double>A,B;
int getNum(char s[]){
int len=strlen(s);
int b=0;
if(s[0]=='-')b=1;
int tmp=1,res=0;
for(int i=len-1;i>=b;i--){
res+=(s[i]-'0')*tmp;
tmp*=10;
}
return res;
}
int main(){
int n,m,a,b;
scanf("%d%d",&n,&m);
for(int i=1;i<=m;i++){
scanf("%d",&t[i]);
for(int j=1;j<=t[i];j++){
scanf("%s",s);
int tmp=getNum(s);
if(s[0]=='-')F[i].insert(tmp);
else M[i].insert(tmp);
}
}
scanf("%s%s",s,s1);
a=getNum(s),b=getNum(s1);
if(s[0]=='-'){//a女的,b男的
for(int i=1;i<=m;i++){
if(F[i].find(a)!=F[i].end()){
for(set<int>::iterator it=M[i].begin();it!=M[i].end();it++){
A[*it]+=1/(double)t[i];
}
}
if(M[i].find(b)!=M[i].end()){
for(set<int>::iterator it=F[i].begin();it!=F[i].end();it++){
B[*it]+=1/(double)t[i];
}
}
}
}
else{//a男的,b女的
for(int i=1;i<=m;i++){
if(M[i].find(a)!=M[i].end()){
for(set<int>::iterator it=F[i].begin();it!=F[i].end();it++){
A[*it]+=1/(double)t[i];
}
}
if(F[i].find(b)!=F[i].end()){
for(set<int>::iterator it=M[i].begin();it!=M[i].end();it++){
B[*it]+=1/(double)t[i];
}
}
}
}
double mx1=0,mx2=0;
for(map<int,double>::iterator it=A.begin();it!=A.end();it++){
mx1=max(mx1,it->second);
}
for(map<int,double>::iterator it=B.begin();it!=B.end();it++){
mx2=max(mx2,it->second);
}
if(A[b]==mx1&&B[a]==mx2){
printf("%s %s\n",s,s1);
}
else{
for(map<int,double>::iterator it=A.begin();it!=A.end();it++){
if(it->second==mx1){
printf("%s ",s);
if(s[0]=='-')printf("%d\n",it->first);
else printf("-%d\n",it->first);
}
}
for(map<int,double>::iterator it=B.begin();it!=B.end();it++){
if(it->second==mx2){
printf("%s ",s1);
if(s[0]=='-')printf("-%d\n",it->first);
else printf("%d\n",it->first);
}
}
}
}