結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー arahi10
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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))
0