Submission #3979777


Source Code Expand

def warshall_floyd(v_count: int, matrix: list) -> list:
    """ ワーシャルフロイド(user: htkb)
    :param v_count: 頂点数
    :param matrix: 隣接行列(到達不能はfloat("inf"))
    """
    # 到達不能をfloat("inf")にしておけば余計なチェックを入れなくても
    # inf > inf+(-1) のような到達不能+負辺が繋がってしまうことはない
    for i in range(v_count):
        for j, c2 in enumerate(row[i] for row in matrix):
            for k, (c1, c3) in enumerate(zip(matrix[j], matrix[i])):
                  matrix[j][k] = min(c1, c2+c3)
    return matrix


if __name__ == "__main__":
    import sys
    N, M = map(int, input().split())

    inf = float("inf")
    matrix = [[inf]*N for _ in [0]*N]
    for i in range(N):
        matrix[i][i] = 0

    for a, b, t in (map(int, l.split()) for l in sys.stdin):
        matrix[a-1][b-1] = matrix[b-1][a-1] = t

    matrix = warshall_floyd(N, matrix)

    print(min(max(row) for row in matrix))

Submission Info

Submission Time
Task D - バスと避けられない運命
User odanny
Language Python (3.4.3)
Score 0
Code Size 1027 Byte
Status TLE
Exec Time 5256 ms
Memory 5208 KB

Judge Result

Set Name All
Score / Max Score 0 / 100
Status
AC × 30
TLE × 9
Set Name Test Cases
All sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt, test_13.txt, test_14.txt, test_15.txt, test_16.txt, test_17.txt, test_18.txt, test_19.txt, test_20.txt, test_21.txt, test_22.txt, test_23.txt, test_24.txt, test_25.txt, test_26.txt, test_27.txt, test_28.txt, test_29.txt, test_30.txt, test_31.txt, test_32.txt, test_33.txt, test_34.txt, test_35.txt, test_36.txt
Case Name Status Exec Time Memory
sample_01.txt AC 18 ms 3064 KB
sample_02.txt AC 18 ms 3064 KB
sample_03.txt AC 18 ms 3064 KB
test_01.txt AC 18 ms 3064 KB
test_02.txt TLE 5256 ms 3956 KB
test_03.txt TLE 5255 ms 5208 KB
test_04.txt AC 2450 ms 4552 KB
test_05.txt AC 1976 ms 3940 KB
test_06.txt AC 193 ms 3188 KB
test_07.txt AC 4974 ms 4796 KB
test_08.txt AC 26 ms 3064 KB
test_09.txt AC 1375 ms 3692 KB
test_10.txt AC 160 ms 3188 KB
test_11.txt AC 188 ms 3188 KB
test_12.txt AC 3702 ms 5056 KB
test_13.txt AC 202 ms 3188 KB
test_14.txt AC 302 ms 3188 KB
test_15.txt AC 4551 ms 4688 KB
test_16.txt AC 347 ms 3316 KB
test_17.txt AC 1621 ms 3804 KB
test_18.txt AC 3207 ms 4912 KB
test_19.txt AC 389 ms 3316 KB
test_20.txt AC 316 ms 3316 KB
test_21.txt AC 18 ms 3064 KB
test_22.txt AC 1078 ms 3792 KB
test_23.txt AC 21 ms 3064 KB
test_24.txt TLE 5256 ms 3700 KB
test_25.txt TLE 5256 ms 3828 KB
test_26.txt AC 1687 ms 4200 KB
test_27.txt AC 2269 ms 4544 KB
test_28.txt TLE 5255 ms 3700 KB
test_29.txt TLE 5256 ms 3700 KB
test_30.txt AC 333 ms 3436 KB
test_31.txt AC 39 ms 3064 KB
test_32.txt TLE 5256 ms 3828 KB
test_33.txt TLE 5256 ms 3700 KB
test_34.txt AC 119 ms 3188 KB
test_35.txt AC 1319 ms 4084 KB
test_36.txt TLE 5256 ms 3700 KB