結果

問題 No.1065 電柱 / Pole (Easy)
ユーザー はむ吉🐹
提出日時 2020-05-29 22:11:13
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 1,334 bytes
コンパイル時間 239 ms
コンパイル使用メモリ 12,672 KB
実行使用メモリ 138,784 KB
最終ジャッジ日時 2024-11-06 05:04:56
合計ジャッジ時間 5,953 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 1 TLE * 1 -- * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#!/usr/bin/env python3

import heapq
import math
 
 
INF = 10 ** 8


def dijkstra(num_vertices, adj_list, source=0):
    dist = [INF for _ in range(num_vertices)]
    dist[source] = 0
    pq = [(dist[u], u) for u in range(num_vertices)]
    heapq.heapify(pq)
    while pq:
        _, u = heapq.heappop(pq)
        for v, cost in adj_list[u]:
            new_length = dist[u] + cost
            if new_length < dist[v]:
                dist[v] = new_length
                heapq.heappush(pq, (new_length, v))
    return dist



def minimize(n, adj_list, pole_start, pole_end):
    dist = dijkstra(n, adj_list, pole_start)
    res = dist[pole_end]
    return res


def main():
    n, m = (int(z) for z in input().split())
    pole_start, pole_end = (int(z) - 1 for z in input().split())
    pole_poss = []
    for _ in range(n):
        x0, y0 = (float(z) for z in input().split())
        pole_poss.append((x0, y0))
    adj_list = [set() for _ in range(n)]
    for _ in range(m):
        st, en = (int(z) - 1 for z in input().split())
        dist = math.hypot(pole_poss[st][0] - pole_poss[en][0], pole_poss[st][1] - pole_poss[en][1])
        adj_list[st].add((en, dist))
        adj_list[en].add((st, dist))
    res = minimize(n, adj_list, pole_start, pole_end)
    print("{:.12f}".format(res))


if __name__ == '__main__':
    main()
0