結果

問題 No.2179 Planet Traveler
ユーザー MasKoaTS
提出日時 2022-09-06 20:55:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 544 ms / 3,000 ms
コード長 987 bytes
コンパイル時間 270 ms
コンパイル使用メモリ 82,432 KB
実行使用メモリ 149,084 KB
最終ジャッジ日時 2024-11-30 16:28:42
合計ジャッジ時間 6,567 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys, math
from heapq import heappop, heappush
input = sys.stdin.readline
INF = 10 ** 18

def isqrt(n):
	rn = math.sqrt(n)
	ok = max(0, int(rn - 2))
	ng = int(rn + 2)
	while(ng - ok > 1):
		k = (ok + ng) >> 1
		if(k * k <= n):
			ok = k
		else:
			ng = k
	return ok

def dijkstra(route):
	path = [INF] * n
	que = [(0, 0)]
	while(que):
		d, v = heappop(que)
		if(path[v] <= d):
			continue
		path[v] = d
		if(v == n - 1):
			break
		for nv, nd in route[v]:
			heappush(que, (max(d, nd), nv))
	return path[n - 1]


"""
Main Code
"""

n = int(input())
planet = [list(map(int, input().split())) for _ in [0] * n]

route = [[] for _ in [0] * n]
for i in range(0, n-1):
	for j in range(i+1, n):
		p1 = planet[i]
		p2 = planet[j]
		d = (p1[0] - p2[0]) ** 2 + (p1[1] - p2[1]) ** 2
		if(p1[2] != p2[2]):
			r1 = p1[0] ** 2 + p1[1] ** 2
			r2 = p2[0] ** 2 + p2[1] ** 2
			d = r1 + r2 - isqrt(4 * r1 * r2)
		route[i].append((j, d))
		route[j].append((i, d))

ans = dijkstra(route)
print(ans)
0