天梯赛L2-044 大众情人（Floyd最短路）

今天看到PTA更新了天梯赛的题目，果断去做了一下当时没写完的题（ps：我是fw，今年的题目也太恶心了叭）

原题链接：传送门

思路

根据题面，要先跑一遍Floyd最短路，注意是有向图，之后按照题意排个序就好啦

题目描述

人与人之间总有一点距离感。我们假定两个人之间的亲密程度跟他们之间的距离感成反比，并且距离感是单向的。例如小蓝对小红患了单相思，从小蓝的眼中看去，他和小红之间的距离为 1，只差一层窗户纸；但在小红的眼里，她和小蓝之间的距离为 108000，差了十万八千里…… 另外，我们进一步假定，距离感在认识的人之间是可传递的。例如小绿觉得自己跟小蓝之间的距离为 2，则即使小绿并不直接认识小红，我们也默认小绿早晚会认识小红，并且因为跟小蓝很亲近的关系，小绿会觉得自己跟小红之间的距离为 1+2=3。当然这带来一个问题，如果小绿本来也认识小红，或者他通过其他人也能认识小红，但通过不同渠道推导出来的距离感不一样，该怎么算呢？我们在这里做个简单定义，就将小绿对小红的距离感定义为所有推导出来的距离感的最小值。

一个人的异性缘不是由最喜欢他/她的那个异性决定的，而是由对他/她最无感的那个异性决定的。我们记一个人 i 在一个异性 j 眼中的距离感为Dij；将 i 的“异性缘”定义为 1/max j∈S(i){Dij}，其中 S(i) 是相对于 i 的所有异性的集合。那么“大众情人”就是异性缘最好（值最大）的那个人。

本题就请你从给定的一批人与人之间的距离感中分别找出两个性别中的“大众情人”。

输入格式

输入在第一行中给出一个正整数 N（≤500），为总人数。于是我们默认所有人从 1 到 N 编号。

随后 N 行，第 i 行描述了编号为 i 的人与其他人的关系，格式为：

性别 K 朋友1:距离1 朋友2:距离2 …… 朋友K:距离K

其中 性别 是这个人的性别，F 表示女性，M 表示男性；K（<N 的非负整数）为这个人直接认识的朋友数；随后给出的是这 K 个朋友的编号、以及这个人对该朋友的距离感。距离感是不超过1e6的正整数。

题目保证给出的关系中一定两种性别的人都有，不会出现重复给出的关系，并且每个人的朋友中都不包含自己。

输出描述

第一行给出自身为女性的“大众情人”的编号，第二行给出自身为男性的“大众情人”的编号。如果存在并列，则按编号递增的顺序输出所有。数字间以一个空格分隔，行首尾不得有多余空格。

输入样例

6
F 1 4:1
F 2 1:3 4:10
F 2 4:2 2:2
M 2 5:1 3:2
M 2 2:2 6:2
M 2 3:1 2:5

输出描述

2 3
4

代码

#include<bits/stdc++.h>

using namespace std;

typedef pair<int, double>PII; //first存编号，second存该编号的异性缘 
int n;
int mp[510][510];
unordered_map<int, int>xb; //xb为1是男性，为2是女性 
vector<PII>man,woman; //分别存男生和女生的异性缘 

bool cmp(PII x, PII y)
{
	if(x.second == y.second) return x.first < y.first; //如果异性缘一样，按编号递增排序 
	return x.second > y.second; //按异性缘降序排序 
}

int main()
{
    memset(mp, 0x3f, sizeof mp); //设初始距离为无穷 
    cin >> n;
    for(int i = 1; i <= n; i ++ )
    {
    	int k;
    	char c;
    	cin >> c >> k;
		for(int j = 1; j <= k; j ++ )
		{
			int x,y;
			scanf("%d:%d", &x, &y);
			mp[i][x] = y; //有向图 
		}
		if(c == 'M') xb[i] = 1;
		else xb[i] = 2;
	}
	 
    for(int i = 1; i <= n; i ++ ) mp[i][i] = 0;
    for(int k = 1; k <= n; k ++ )
    {
        for(int i = 1; i <= n; i ++ )
        {
            for(int j = 1; j <= n; j ++ )
            {
                mp[i][j] = min(mp[i][j], mp[i][k] + mp[k][j]);
            }
        }
    }
    //以上为Floyd最短路的模板
	 
    for(int i = 1; i <= n; i ++ ) //i = 1开始遍历，遍历每个异性和i的距离 
    {
    	int maxx = -1;
    	for(int j = 1; j <= n; j ++ )
    	{
    		if(i != j && xb[i] != xb[j]) //如果是异性就取距离最大值 
    		{
    			maxx = max(maxx, mp[j][i]);
			}
		}
		if(xb[i] == 1) man.push_back({i, 1.0 / maxx});
		else woman.push_back({i, 1.0 / maxx});
	}
	sort(man.begin(), man.end(), cmp);
	sort(woman.begin(), woman.end(), cmp);
	//按题目要求输出即可 
	cout << woman[0].first;
	for(int i = 1; i < woman.size(); i ++ )
	{
		if(woman[i].second == woman[i - 1].second) cout << " " << woman[i].first;
		else break;
	}
	cout << '\n';
	cout << man[0].first;
	for(int i = 1; i < man.size(); i ++ )
	{
		if(man[i].second == man[i - 1].second) cout << " " << man[i].first;
		else break;
	}
    return 0;
}