結果
問題 |
No.1170 Never Want to Walk
|
ユーザー |
|
提出日時 | 2020-08-14 21:49:48 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,449 bytes |
コンパイル時間 | 570 ms |
コンパイル使用メモリ | 82,232 KB |
実行使用メモリ | 108,816 KB |
最終ジャッジ日時 | 2024-10-10 14:59:56 |
合計ジャッジ時間 | 7,677 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 TLE * 1 -- * 6 |
ソースコード
n,a,b=map(int,input().split()) class UnionFind: #def -> foo=UnionFind(n,1) <- 1-based index, default is 0 #method -> foo.hoge(huga) __slots__ = ["_size", "_first_idx", "_parents"] def __init__(self, size: int, first_index: int = 0) -> None: self._size = size self._first_idx = first_index self._parents = [-1] * (size + first_index) def find(self, x: int) -> int: if self._parents[x] < 0: return x self._parents[x] = self.find(self._parents[x]) return self._parents[x] def same(self, x: int, y: int) -> bool: return self.find(x) == self.find(y) def unite(self, x: int, y: int) -> bool: x, y = self.find(x), self.find(y) if x == y: return False if self._parents[x] > self._parents[y]: x, y = y, x self._parents[x] += self._parents[y] self._parents[y] = x return True def size(self, x: int) -> int: return -self._parents[self.find(x)] def group_count(self) ->int: return sum(1 for i in self._parents if i<0)-self._first_idx def connected(self) ->bool: return self._parents[self.find(self._first_idx)]==-self._size l=list(map(int,input().split())) ue=0 uf=UnionFind(n) from bisect import bisect_left as bl,bisect_right as br for i,x in enumerate(l): for j in range(bl(l,x-b),br(l,x-a)):uf.unite(i,j) for i in range(n):print(uf.size(i))