AtCoder ABC 304 C - Virus を PyPy3とPythonでDFS, BFS, Union-Find, (ダイクストラ法) などで解く - Qiita

import sys sys.setrecursionlimit(100000) n, d = map(int, input().split()) dat = [] for _ in range(n): dat.append(tuple(map(int, input().split()))) visited = [False] * n def func(cur): x1, y1 = dat[cur] for i in range(n): # 全探索 if visited[i]: continue # 既に感染 x2, y2 = dat[i] if (x1-x2)**2 + (y1-y2)**2 <= (d**2): visited[i] = True func(i) visited[0] = True func(0) for x in visited: print("Yes" if x e