結果

問題 No.2179 Planet Traveler
コンテスト
ユーザー ニックネーム
提出日時 2023-01-07 00:28:01
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,308 ms / 3,000 ms
コード長 949 bytes
コンパイル時間 306 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 126,760 KB
最終ジャッジ日時 2024-11-30 21:31:10
合計ジャッジ時間 17,824 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

class UF:
    def __init__(self,n):
        self.p = [-1]*n
    def f(self,x):
        if self.p[x]<0: return x
        else: self.p[x] = self.f(self.p[x]); return self.p[x]
    def u(self,x,y):
        x = self.f(x); y = self.f(y)
        if x==y: return
        if -self.p[x]<-self.p[y]: x,y = y,x
        self.p[x] += self.p[y]; self.p[y] = x
    def s(self,x,y): return self.f(x) == self.f(y)
def isqrt(n):
    m = int(n**0.5)
    if m**2>n: m -= 1
    if (m+1)**2<=n: m += 1
    return m
n = int(input())
xyt = [tuple(map(int,input().split())) for _ in range(n)]
dij = []
for i,(xi,yi,ti) in enumerate(xyt):
    for j,(xj,yj,tj) in enumerate(xyt):
        if i>=j: continue
        if ti==tj: d = (xi-xj)**2+(yi-yj)**2
        else: d = (xi**2+yi**2+xj**2+yj**2)-isqrt(4*(xi**2+yi**2)*(xj**2+yj**2))
        dij.append((d,i,j))
uf = UF(n)
s = 0
for d,i,j in sorted(dij):
    if not uf.s(i,j): uf.u(i,j); s = d
    if uf.s(0,n-1): break
print(s)
0