没能做出来,两个点没过
正确代码: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;
}