PAT1 1072 Gas Station

题目链接
我的github

题目大意

有一些居民地和气站，并给出他们之间的距离。现在要从这些气站中选出一个，使得选出的这个气站与所有居民地的距离最小值最大，且不超过气站的服务范围

输入

每组包含一个测试用例

• 第一行是四个正整数$N\le 10^3$表示居民地的数量，$M\le 10$表示气站的数量，$K\le 10^4$表示道路的数量，$D_S$表示气站的服务范围
• 之后有$K$行表示道路，格式为P1 P2 Dist，P1 P2表示两个地点，Dist表示这两个地点之间的距离

居民地的编号从$1$~$N$，气站的编号从$G1$~$GM$

输出

对每个测试用例，现在一行中输出气站的编号，在第二行中输出气站与居民地的最小距离和平均值。
如果最小距离不唯一就选出平均值最小的那个，如果还是不唯一就选出编号最小的那个

样例输入

4 3 11 5
1 2 2
1 4 2
1 G1 4
1 G2 3
2 3 2
2 G2 1
3 4 2
3 G3 2
4 G1 3
G2 G1 1
G3 G2 2

2 1 2 10
1 G1 9
2 G1 20

样例输出

G1
2.0 3.3

No Solution

解析

这题使用$M$次dijkstra算法分别求出每个气站与居民地的最小距离，然后从这些最小距离中选取最大的值，如果不唯一，就再找平均距离最小的。由于我们是按照气站编号顺序查找的，所以这样找出来的必定是符合输出要求的

但是这题使用pyhton会有个测试点超时

所以用C++再写一遍就行，参考：传送门
超时的python：

# -*- coding: utf-8 -*- 
# @Time : 2019/6/30 14:11 
# @Author : ValarMorghulis 
# @File : 1072.py
INF = 0x3f3f3f3f


def solve():
    n, m, k, ds = map(int, input().split())
    edge = [[INF for i in range(n + m + 5)] for j in range(n + m + 5)]
    for i in range(k):
        a, b, c = input().split()
        c = int(c)
        if a[0] == 'G':
            a = n + int(a[1])
        else:
            a = int(a)
        if b[0] == 'G':
            b = n + int(b[1])
        else:
            b = int(b)
        edge[a][b] = edge[b][a] = c
    ansId, ansDis, ansAvg = -1, -1, INF
    for i in range(n + 1, n + m + 1):
        minDis, avg = INF, 0
        dis = [INF for i in range(n + m + 5)]
        vis = [False for i in range(n + m + 5)]
        dis[i] = 0
        for j in range(n + m):
            mark, md = -1, INF
            for x in range(1, n + m + 1):
                if not vis[x] and dis[x] < md:
                    mark, md = x, dis[x]
            if mark == -1:
                break
            vis[mark] = True
            for x in range(1, n + m + 1):
                if not vis[x] and dis[x] > dis[mark] + edge[mark][x]:
                    dis[x] = dis[mark] + edge[mark][x]
        for j in range(1, n + 1):
            if dis[j] > ds:
                minDis = -1
                break
            if dis[j] < minDis:
                minDis = dis[j]
            avg += dis[j]
        if minDis == -1:
            continue
        avg /= n
        if minDis > ansDis:
            ansId, ansDis, ansAvg = i, minDis, avg
        elif minDis == ansDis and ansAvg > avg:
            ansId, ansAvg = i, avg
    if ansId == -1:
        print("No Solution")
    else:
        print("G%d\n%.1f %.1f" % (ansId-n, ansDis, ansAvg))


if __name__ == "__main__":
    solve()