結果

問題 No.1170 Never Want to Walk
ユーザー shobonvipshobonvip
提出日時 2020-08-14 22:58:58
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 2,098 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 82,596 KB
実行使用メモリ 108,756 KB
最終ジャッジ日時 2024-10-10 16:19:44
合計ジャッジ時間 8,563 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
58,112 KB
testcase_01 AC 37 ms
52,608 KB
testcase_02 AC 38 ms
52,992 KB
testcase_03 AC 38 ms
52,992 KB
testcase_04 AC 38 ms
52,864 KB
testcase_05 AC 38 ms
52,608 KB
testcase_06 AC 38 ms
53,120 KB
testcase_07 AC 37 ms
52,736 KB
testcase_08 AC 37 ms
52,736 KB
testcase_09 AC 39 ms
52,352 KB
testcase_10 AC 39 ms
52,352 KB
testcase_11 AC 40 ms
52,736 KB
testcase_12 AC 67 ms
71,424 KB
testcase_13 AC 81 ms
76,160 KB
testcase_14 AC 80 ms
75,936 KB
testcase_15 AC 66 ms
72,384 KB
testcase_16 AC 69 ms
73,156 KB
testcase_17 AC 85 ms
76,536 KB
testcase_18 AC 66 ms
70,656 KB
testcase_19 AC 68 ms
73,600 KB
testcase_20 AC 90 ms
76,412 KB
testcase_21 AC 68 ms
71,040 KB
testcase_22 AC 91 ms
76,672 KB
testcase_23 AC 91 ms
76,288 KB
testcase_24 AC 94 ms
76,296 KB
testcase_25 AC 91 ms
76,800 KB
testcase_26 AC 91 ms
76,416 KB
testcase_27 AC 431 ms
101,784 KB
testcase_28 AC 408 ms
101,560 KB
testcase_29 AC 452 ms
102,788 KB
testcase_30 AC 434 ms
101,888 KB
testcase_31 AC 444 ms
102,000 KB
testcase_32 TLE -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

import bisect
class UnionFind:
	"""
	class UnionFind
	https://note.nkmk.me/python-union-find/
	
	order O(log(n))
	
	n = the number of elements
	parents = the list that contains "parents" of x. be careful that it doesn't contain "root" of x.
	"""
	def __init__(self, n):
		"""
		make UnionFind
		:param n: the number of elements
		"""
		self.n = n
		self.parents = [-1] * n
	
	def find(self, x):
		"""
		:param x: an element
		:return: the root containing 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):
		"""
		:param x, y: an element
		"""
		# root
		x = self.find(x)
		y = self.find(y)
		if x == y:
			# already same group so do nothing
			return
		if self.parents[x] > self.parents[y]:
			# the root should be min of group
			x, y = y, x
		# remind that x, y is the root, x < y, then, y unions to x.
		self.parents[x] += self.parents[y]
		# and then y's parent is x.
		self.parents[y] = x
	def size(self, x):
		# return the size of group
		return -self.parents[self.find(x)]
	
	def same(self, x, y):
		# return whether the x, y are in same group
		return self.find(x) == self.find(y)
	
	def members(self, x):
		# return members of group
		root = self.find(x)
		return [i for i in range(self.n) if self.find(i) == root]
	
	def roots(self):
		# return all roots
		return [i for i, x in enumerate(self.parents) if x < 0]
	
	def group_count(self):
		# return how many groups
		return len(self.roots())
	
	def all_group_members(self):
		# return all members of all groups
		return {r: self.members(r) for r in self.roots()}

n,a,b = map(int,input().split())
x = list(map(int,input().split()))
x.sort()
union = UnionFind(n)

for i in range(n):
	targ1 = bisect.bisect_right(x, x[i]+b)
	targ2 = bisect.bisect_left(x, x[i]+a)
	targ3 = bisect.bisect_right(x, x[i]-a)
	targ4 = bisect.bisect_left(x, x[i]-b)
	if targ1 - targ2 > 0:
		for j in range(targ2, targ1):
			union.union(i, j)
	if targ3 - targ4 > 0:
		for j in range(targ4, targ3):
			union.union(i, j)

for i in range(n):
	print(union.size(i))
0