結果
問題 | No.1265 Balloon Survival |
ユーザー |
![]() |
提出日時 | 2022-05-30 23:22:27 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,438 bytes |
コンパイル時間 | 159 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 168,704 KB |
最終ジャッジ日時 | 2024-09-21 00:56:16 |
合計ジャッジ時間 | 9,812 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 27 TLE * 5 |
ソースコード
import sys import pypyjit import itertools import heapq import math from collections import deque, defaultdict import bisect input = sys.stdin.readline sys.setrecursionlimit(10 ** 6) pypyjit.set_param('max_unroll_recursion=-1') def index_lt(a, x): 'return largest index s.t. A[i] < x or -1 if it does not exist' return bisect.bisect_left(a, x) - 1 def index_le(a, x): 'return largest index s.t. A[i] <= x or -1 if it does not exist' return bisect.bisect_right(a, x) - 1 def index_gt(a, x): 'return smallest index s.t. A[i] > x or len(a) if it does not exist' return bisect.bisect_right(a, x) def index_ge(a, x): 'return smallest index s.t. A[i] >= x or len(a) if it does not exist' return bisect.bisect_left(a, x) N = int(input()) C = [tuple(map(int, input().split())) for _ in range(N)] class Dist: def __init__(self, d, x, y): self.d = d self.pair = (x, y) def __lt__(self, other): return self.d < other.d h = [] for i in range(N): for j in range(i + 1, N): xi, yi = C[i] xj, yj = C[j] d = (xi - xj) ** 2 + (yi - yj) ** 2 heapq.heappush(h, Dist(d, i, j)) S = {i for i in range(N)} ans = 0 while len(S) > 1: dist = heapq.heappop(h) a, b = dist.pair if a in S and b in S: if a == 0: S.remove(b) ans += 1 else: S.remove(a) S.remove(b) print(ans)