Submission #3917007

Source Code Expand

Copy
def warshall_floyd(v_count: int, matrix: list) -> list:
    """ ワーシャルフロイド
    :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])):
                if c1 > c2+c3:
                    matrix[j][k] = 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 htkb
Language Python3 (3.4.3)
Score 100
Code Size 1038 Byte
Status
Exec Time 3049 ms
Memory 6564 KB

Test Cases

Set Name Score / Max Score Test Cases
All 100 / 100 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 17 ms 3064 KB
sample_02.txt 17 ms 3064 KB
sample_03.txt 17 ms 3064 KB
test_01.txt 17 ms 3064 KB
test_02.txt 3049 ms 6500 KB
test_03.txt 2637 ms 5172 KB
test_04.txt 1001 ms 4656 KB
test_05.txt 776 ms 3940 KB
test_06.txt 87 ms 3188 KB
test_07.txt 1910 ms 4800 KB
test_08.txt 21 ms 3064 KB
test_09.txt 528 ms 3680 KB
test_10.txt 75 ms 3188 KB
test_11.txt 85 ms 3188 KB
test_12.txt 1445 ms 5032 KB
test_13.txt 90 ms 3188 KB
test_14.txt 130 ms 3316 KB
test_15.txt 1741 ms 4680 KB
test_16.txt 148 ms 3316 KB
test_17.txt 599 ms 3800 KB
test_18.txt 1285 ms 4892 KB
test_19.txt 160 ms 3316 KB
test_20.txt 138 ms 3316 KB
test_21.txt 17 ms 3064 KB
test_22.txt 439 ms 3784 KB
test_23.txt 19 ms 3064 KB
test_24.txt 2854 ms 6504 KB
test_25.txt 2844 ms 6540 KB
test_26.txt 634 ms 4312 KB
test_27.txt 837 ms 4528 KB
test_28.txt 2875 ms 6552 KB
test_29.txt 2888 ms 6564 KB
test_30.txt 139 ms 3432 KB
test_31.txt 26 ms 3064 KB
test_32.txt 2863 ms 6564 KB
test_33.txt 2875 ms 6528 KB
test_34.txt 56 ms 3188 KB
test_35.txt 494 ms 4072 KB
test_36.txt 2849 ms 6560 KB