結果

問題 No.2923 Mayor's Job
ユーザー manuo
提出日時 2024-10-12 15:02:22
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 525 ms / 2,000 ms
コード長 1,214 bytes
コンパイル時間 175 ms
コンパイル使用メモリ 82,076 KB
実行使用メモリ 153,220 KB
最終ジャッジ日時 2024-10-12 15:02:27
合計ジャッジ時間 4,878 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

from collections import deque, defaultdict, Counter
from bisect import bisect_left, bisect_right
from itertools import permutations, combinations
from functools import cmp_to_key, cache
from heapq import heappop, heappush
import math, sys
import pypyjit
pypyjit.set_param('max_unroll_recursion=-1')
sys.setrecursionlimit(10**7)
_int = lambda x: int(x)-1
MOD = 998244353
INF = 1<<62
Yes, No = "Yes", "No"

N, K = map(int, input().split())
H = list(map(int, input().split()))

XY = []
for i in range(N):
    x, y = map(int, input().split())
    XY.append((x, y))

def ok(i, j):
    xi, yi = XY[i]
    xj, yj = XY[j]
    dist = (xi-xj)**2 + (yi-yj)**2
    return dist <= K*K

E = defaultdict(list)
for i in range(N):
    for j in range(i+1, N):
        if ok(i, j):
            if H[i] > H[j]:
                E[i].append(j)
            elif H[i] < H[j]:
                E[j].append(i)

Q = [(H[i], i) for i in range(N)]
Q.sort(reverse=True)
memo = [-1]*N
for h, start in Q:
    if memo[start] != -1: continue
    memo[start] = 1
    q = deque([start])
    while q:
        i = q.popleft()
        for j in E[i]:
            if memo[j] != -1: continue
            memo[j] = 0
            q.append(j)
print(sum(memo))
0