結果
問題 | No.2179 Planet Traveler |
ユーザー |
![]() |
提出日時 | 2023-01-06 22:10:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 325 ms / 3,000 ms |
コード長 | 1,277 bytes |
コンパイル時間 | 967 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 101,760 KB |
最終ジャッジ日時 | 2024-11-30 18:19:02 |
合計ジャッジ時間 | 6,120 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 26 |
ソースコード
class UnionFind: def __init__(self, n): self.n = n self.parents = [-1] * n def find(self, x): if self.parents[x] < 0: return x else: self.parents[x] = self.find(self.parents[x]) return self.parents[x] def union(self, x, y): x = self.find(x) y = self.find(y) if x == y: return if self.parents[x] > self.parents[y]: x, y = y, x self.parents[x] += self.parents[y] self.parents[y] = x n = int(input()) x = [0] * n y = [0] * n t = [0] * n uf = UnionFind(n) for i in range(n): x[i], y[i], t[i] = map(int,input().split()) a = [] dist = 0 for i in range(n): for j in range(i+1, n): if t[i] == t[j]: dist = (x[i] - x[j]) ** 2 + (y[i] - y[j]) ** 2 else: p2 = x[i] ** 2 + y[i] ** 2 q2 = x[j] ** 2 + y[j] ** 2 p2, q2 = min(p2, q2), max(p2, q2) suki = 10 ** 12 kirai = -1 while suki - kirai > 1: targ = (suki + kirai) // 2 tmp = q2 - p2 - targ if tmp <= 0 or tmp * tmp <= 4 * targ * p2: suki = targ else: kirai = targ dist = suki #print(dist) a.append((dist << 30) + (i << 15) + j) a.sort() mask = (1 << 15) - 1 ans = 0 for v in a: dist = v >> 30 i = (v >> 15) & mask j = v & mask ans = dist #print(dist, i, j) uf.union(i, j) if uf.find(0) == uf.find(n-1): break print(ans)