結果

問題 No.1065 電柱 / Pole (Easy)
ユーザー Fukumoto
提出日時 2020-06-08 19:49:00
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,805 ms / 2,000 ms
コード長 985 bytes
コンパイル時間 241 ms
コンパイル使用メモリ 12,416 KB
実行使用メモリ 119,936 KB
最終ジャッジ日時 2024-12-26 05:14:28
合計ジャッジ時間 37,972 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 46
権限があれば一括ダウンロードができます

ソースコード

diff #

from heapq import heappush, heappop, heapify
import sys
input = sys.stdin.readline

def main():
    N, M = list(map(int, input().split()))
    X, Y = list(map(int, input().split()))
    X -= 1; Y -= 1
    pq = [list(map(int, input().split())) for i in range(N)]
    g = [[] for i in range(N)]
    INF = float('inf')
    dist = [INF for i in range(N)]
    visited = [False] * N

    for i in range(M):
        p, q = list(map(int, input().split()))
        p -= 1; q -= 1

        px, py = pq[p]
        qx, qy = pq[q]

        d = ((px-qx)**2 + (py-qy)**2) **0.5
        g[p].append((d, q))
        g[q].append((d, p))

    q = g[X]
    heapify(q)
    dist[X] = 0
    visited[X] = True

    while q:
        c, v = heappop(q)

        if visited[v]: continue
        visited[v] = True
        dist[v] = c
        if v == Y:
            print(c)
            break

        for d, u in g[v]:
            if not visited[u]: heappush(q, (c+d, u))

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