L2-044 大众情人(Python3)

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

一个人的异性缘不是由最喜欢他/她的那个异性决定的，而是由对他/她最无感的那个异性决定的。我们记一个人 i 在一个异性 j 眼中的距离感为 Dij​；将 i 的“异性缘”定义为 1/maxj∈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 个朋友的编号、以及这个人对该朋友的距离感。距离感是不超过 106 的正整数。

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

输出格式：

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

输入样例：

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

提交结果：

代码（测试点3，测试点4运行超时）：

import sys


def floyd():
    for k in range(1, n + 1):
        for i in range(1, n + 1):
            for j in range(1, n + 1):
                dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j])


INF = 1e9
n = int(sys.stdin.readline())
dist = [[0 for i in range(n + 1)] for j in range(n + 1)]
st = [0 for i in range(n + 1)]
d = [0 for i in range(n + 1)]
for i in range(1, n + 1):
    for j in range(1, n + 1):
        if i == j:
            dist[i][j] = 0
        else:
            dist[i][j] = INF
for i in range(1, n + 1):
    data = sys.stdin.readline().split()
    if data[0] == 'M':
        st[i] = 1
    for j in range(int(data[1])):
        a = int(data[j + 2].split(":")[0])
        b = int(data[j + 2].split(":")[1])
        dist[i][a] = min(dist[i][a], b)
floyd()
for i in range(1, n + 1):
    for j in range(1, n + 1):
        if st[i] + st[j] == 1:
            d[i] = max([d[i], dist[j][i]])
ans1 = INF
ans2 = INF
q1 = []
q2 = []
for i in range(1, n + 1):
    if not st[i]:
        if ans1 > d[i]:
            ans1 = d[i]
            q1 = []
            q1.append(i)
        elif ans1 == d[i]:
            q1.append(i)
    else:
        if ans2 > d[i]:
            ans2 = d[i]
            q2 = []
            q2.append(i)
        elif ans2 == d[i]:
            q2.append(i)
for i in range(len(q1)):
    if i != len(q1) - 1:
        print(q1[i], end=" ")
    else:
        print(q1[i])
for i in range(len(q2)):
    if i != len(q2) - 1:
        print(q2[i], end=" ")
    else:
        print(q2[i])