L2-044 大众情人

一开始看错题了，以为是要传递就是加一，然后就不会了，后来才发现就是直接相加就行了

有一个坑点，是异性对这个人的距离，而不是这个人对于异性的距离，这个要注意

#include <bits/stdc++.h>
using namespace std;
#define vi vector<int>
#define sz(x) (int)x.size()
//2025.3.23
int arr[1010][1010];
vi sex(10010);
vi d(10010,0);
void solve(){
    int n;cin>>n;
    for(int i=0;i<=700;i++){
        for(int j=0;j<=700;j++){
            arr[i][j]=1e9;
            if(i==j)arr[i][j]=0;
        }
    }
    for(int i=1;i<=n;i++){
        char op;int k;cin>>op>>k;
        int x,y;
        if(op=='F')sex[i]=1;//女
        else sex[i]=2;//男
        while(k--){
            scanf("%d:%d",&x,&y);
            arr[i][x]=y;
        }
    }
    for(int k=1;k<=n;k++){//floyd
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                arr[i][j]=min(arr[i][j],arr[i][k]+arr[k][j]);
            }
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            if(sex[i]!=sex[j]){
                d[i]=max(d[i],arr[j][i]);//这里一定要是arr[j][i],因为是到这个的距离最远
            }
        }
    }
    vi ans1,ans2;
    int d1=1e9+10,d2=1e9;
    for(int i=1;i<=n;i++){
        if(sex[i]==1){
            if(d[i]<d1){
                d1=d[i];
                ans1.clear();
                ans1.push_back(i);
            }else if(d[i]==d1){
                ans1.push_back(i);
            }
        }else{
            if(d[i]<d2){
                d2=d[i];
                ans2.clear();
                ans2.push_back(i);
            }else if(d[i]==d2){
                ans2.push_back(i);
            }
        }
    }for(int i=0;i<sz(ans1);i++){
        cout<<ans1[i];
        if(i!=sz(ans1)-1){
            cout<<" ";
        }
    }cout<<endl;

    for(int i=0;i<sz(ans2);i++){
        cout<<ans2[i];
        if(i!=sz(ans2)-1){
            cout<<" ";
        }
    }
    

}

signed main()
{
    solve();
    return 0;
}