結果

問題 No.2923 Mayor's Job
ユーザー manuomanuo
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 50 ms
56,240 KB
testcase_01 AC 49 ms
57,068 KB
testcase_02 AC 48 ms
56,276 KB
testcase_03 AC 263 ms
77,456 KB
testcase_04 AC 525 ms
153,220 KB
testcase_05 AC 279 ms
78,208 KB
testcase_06 AC 291 ms
77,736 KB
testcase_07 AC 329 ms
78,876 KB
testcase_08 AC 286 ms
79,060 KB
testcase_09 AC 288 ms
78,028 KB
testcase_10 AC 290 ms
78,948 KB
testcase_11 AC 491 ms
139,376 KB
testcase_12 AC 353 ms
111,064 KB
testcase_13 AC 111 ms
79,056 KB
testcase_14 AC 71 ms
76,848 KB
testcase_15 AC 48 ms
57,180 KB
testcase_16 AC 48 ms
56,384 KB
testcase_17 AC 47 ms
56,688 KB
testcase_18 AC 47 ms
56,660 KB
testcase_19 AC 50 ms
56,536 KB
権限があれば一括ダウンロードができます

ソースコード

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