L2-028 秀恩爱分得快

没能做出来,两个点没过
正确代码:https://www.cnblogs.com/8023spz/p/9534865.html
人家的代码又简单又快

我的垃圾代码(没过)

#include <bits/stdc++.h>
using namespace std;
#define MX 1007
#define QAQ(vec) for(auto&it:vec){cout<<it<<" "<<endl;}cout<<endl;
int N,M,step,a,b,pos=1;
string s;
bool sex[MX];//0->male  1->female
vector<unordered_set<int>> pho;
vector<pair<int,double>> ar,br;
unordered_set<int> s_resa,s_resb;
vector<int>resa,resb;
unordered_map<int,double> am,bm;
void add(int x,unordered_map<int,double>& mp){
    for(int i=0;i<M;i++){
        if(pho[i].find(x)!=pho[i].end()){
            double K=pho[i].size();
            for(auto&it:pho[i]){
                if(sex[it]^sex[x]){
                    mp[it]+=1.0/K;
                }
            }
        }
    }
}
string print(int x){//输出格式
    string ot="";
    if(sex[x]){
        ot="-";
    }
    ot+=to_string(x);
    return ot;
}
int main(){
    cin>>N>>M;
    memset(sex,0,N);//初始化sex数组
    pho=vector<unordered_set<int>>(M);
    for(int i=0;i<M;i++){//输入性别和编号(正的)
        cin>>step;
        for(int j=0;j<step;j++){
            cin>>s;
            if(s[0]=='-') {
                a=-stoi(s);
                sex[a]= true;
            }else{
                a=stoi(s);
            }
            pho[i].insert(a);
        }
    }
    cin>>s;
    if(s[0]=='-') {
        a=-stoi(s);
        sex[a]= true;
    }else{
        a=stoi(s);
    }
    cin>>s;
    if(s[0]=='-') {
        b=-stoi(s);
        sex[b]= true;
    }else{
        b=stoi(s);
    }
    add(a,am);
    add(b,bm);
    //map->vector排序
    for(auto&it:am){
        ar.emplace_back(it.first,it.second);
    }
    for(auto&it:bm){
        br.emplace_back(it.first,it.second);
    }
    sort(ar.begin(),ar.end(),[&](const pair<int,double>v,const pair<int,double> u){
        return v.second==u.second? v.first<u.first:v.second>u.second;
    });
    sort(br.begin(),br.end(),[&](const pair<int,double>v,const pair<int,double> u){
        return v.second==u.second? v.first<u.first:v.second>u.second;
    });
    if(ar.size()>0){
        s_resa.insert(ar[0].first);
        resa.emplace_back(ar[0].first);
        while(pos<ar.size()&&ar[pos].second==ar[pos-1].second){
            s_resa.insert(ar[pos].first);
            resa.emplace_back(ar[pos].first);
            pos++;
        }
    }
    pos=1;
    if(br.size()>0){
        s_resb.insert(br[0].first);
        resb.emplace_back(br[0].first);
        while(pos<ar.size()&&br[pos].second==br[pos-1].second){
            s_resb.insert(br[pos].first);
            resb.emplace_back(br[pos].first);
            pos++;
        }
    }
    if(s_resa.find(b)!=s_resa.end()&&s_resb.find(a)!=s_resb.end()){
        cout<<print(a)<<" "<<print(b)<<endl;
    }else{
        for(auto&it:resa){
            cout<<print(a)<<" "<<print(it)<<endl;
        }
        for(auto&it:resb){
            cout<<print(b)<<" "<<print(it)<<endl;
        }
    }
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值