結果

問題 No.1170 Never Want to Walk
ユーザー trineutron
提出日時 2020-08-15 15:40:36
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
RE  
実行時間 -
コード長 869 bytes
コンパイル時間 245 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 33,040 KB
最終ジャッジ日時 2024-10-10 17:58:11
合計ジャッジ時間 11,105 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30 RE * 7
権限があれば一括ダウンロードができます

ソースコード

diff # 

class UnionFind:
    par = []
    
    def __init__(self, n):
        self.par = [i for i in range(n)]
    
    def find(self, n):
        if self.par[n] == n:
            return n
        p = self.find(self.par[n])
        self.par[n] = p
        return p
    
    def merge(self, m, n):
        m = self.find(m)
        n = self.find(n)
        self.par[m] = n

n, a, b = map(int, input().split())
x = list(map(int, input().split()))
v = [0] * (n + 1)
l = 0
r = 0
t = UnionFind(n)
for i in range(n):
    while l < n and x[l] < x[i] + a:
        l += 1
    while r < n and x[r] <= x[i] + b:
        r += 1
    if l == r:
        continue
    v[l] += 1
    v[r - 1] -= 1
    t.merge(i, l)
for i in range(n):
    v[i + 1] += v[i]
    if v[i] > 0:
        t.merge(i, i + 1)
s = [0] * n
for i in range(n):
    s[t.find(i)] += 1
for i in range(n):
    print(s[t.find(i)])
0