L2-4 秀恩爱分得快 (25 分)

题目链接

输入样例 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 <iostream>
#include<bits/stdc++.h>
#define ll long long
#define PII pair<int,int>
#define pb push_back
#define mem(a,b) memset(a,b,sizeof(a))
const int N = 1e3 + 10;
using namespace std;

int sex[N]; ///性别
double num[N][N]; ///亲密度
double maxn[N]; /// 两人之间最大亲密度
set<int>ma[N]; ///拥有最大值的人
set<int>st[N]; ///记录照片信息

int change(string s){
    bool flag=true;
    if(s[0]=='-'){
        s=s.substr(1);
        flag=false;
    }
    int t=0;
    for(int i=0;i<s.length();i++)
        t=t*10+s[i]-48;
    if(flag)
        sex[t]=1;
    return t;
}

int main(){
    mem(sex,-1);
    int n,m;
    cin>>n>>m;
    for(int i=0;i<m;i++){
        int k;
        cin>>k;
        for(int j=0;j<k;j++){
            string s1;
            cin>>s1;
            int a=change(s1);
            st[i].insert(a);
        }
    }

    int a,b;
    cin>>a>>b;

    a=abs(a);
    b=abs(b);

     for(int i=0;i<m;i++){
        set<int>::iterator it;
        if(st[i].count(a)){
            for(it=st[i].begin();it!=st[i].end();it++){
                if(*it!=a){
                    if(sex[*it]*sex[a]<0){
                        num[a][*it]+=1.0/st[i].size();
                        if(num[a][*it]>maxn[a]){
                            ma[a].clear();
                            maxn[a]=num[a][*it];
                            ma[a].insert(*it);
                        }if(num[a][*it]==maxn[a])
                            ma[a].insert(*it);
                    }
                }
            }
        }
        if(st[i].count(b)){
            for(it=st[i].begin();it!=st[i].end();it++){
                if(*it!=b){
                    if(sex[*it]*sex[b]<0){
                        num[b][*it]+=1.0/st[i].size();
                        if(num[b][*it]>maxn[b]){
                            ma[b].clear();
                            maxn[b]=num[b][*it];
                            ma[b].insert(*it);
                        }if(num[b][*it]==maxn[b])
                            ma[b].insert(*it);
                    }
                }
            }
        }
    }


    if(num[a][b]==maxn[a]&&num[a][b]==maxn[b]){
        if(a*b||sex[0]==1)
            cout<<sex[a]*a<<" "<<sex[b]*b;
        if(a==0&&sex[0]==-1)
            cout<<"-0 "<<b;
        if(b==0&&sex[0]==-1)
            cout<<a<<" -0";
    }else{
        set<int>::iterator it;
        for(it=ma[a].begin();it!=ma[a].end();it++){
            if(*it*a||sex[0]==1)
                cout<<a*sex[a]<<" "<<*it*sex[*it]<<endl;
            if(a==0&&sex[0]==-1)
                cout<<"-0 "<<*it<<endl;
            if(*it==0&&sex[0]==-1)
                cout<<a<<" -0"<<endl;
        }
        set<int>::iterator it1;
        for(it1=ma[b].begin();it1!=ma[b].end();it1++){
            if(*it1*b||sex[0]==1)
                cout<<b*sex[b]<<" "<<*it1*sex[*it1]<<endl;
            if(b==0&&sex[0]==-1)
                cout<<"-0 "<<*it1<<endl;
            if(*it1==0&&sex[0]==-1)
                cout<<b<<" -0"<<endl;
        }
    }
    return 0;
}