結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー arahi10
提出日時 2020-08-14 22:00:47
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,548 bytes
コンパイル時間 246 ms
コンパイル使用メモリ 82,324 KB
実行使用メモリ 108,700 KB
最終ジャッジ日時 2024-10-10 15:15:12
合計ジャッジ時間 7,387 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30 TLE * 1 -- * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

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()))
uf=UnionFind(n)
from bisect import bisect_left as bl,bisect_right as br
left,right=0,0
for i,x in enumerate(l):
    while left <n and  l[left]<x-b:left+=1
    while right <n and l[right]<=x-a:right+=1
    for j in range(left,right):uf.unite(i,j)
for i in range(n):print(-uf._parents[uf.find(i)])
0