結果

問題 No.1170 Never Want to Walk
コンテスト
ユーザー Kiri8128
提出日時 2020-08-14 22:59:44
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 865 bytes
コンパイル時間 163 ms
コンパイル使用メモリ 82,176 KB
実行使用メモリ 103,684 KB
最終ジャッジ日時 2024-10-10 16:20:01
合計ジャッジ時間 6,680 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 21 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

from bisect import bisect_right as br
def par(a):
    L = []
    while P[a] >= 0:
        L.append(a)
        a = P[a]
    for l in L:
        P[l] = a
    return a
def unite(a, b):
    if par(a) != par(b):
        if P[par(b)] >= P[par(a)]:
            if P[par(b)] == P[par(a)]: P[par(a)] -= 1
            P[par(b)] = par(a)
        else:
            P[par(a)] = par(b)

N, A, B = map(int, input().split())
P = [-1 for i in range(N)]
X = [int(a) for a in input().split()]
done = [0] * N
i = 0
for i in range(N):
    if done[i]: continue
    done[i] = 1
    j0 = br(X, X[i] + B) - 1
    j = j0
    while X[j] >= X[i] + A:
        unite(i, j)
        done[j] = 1
        j -= 1
        if done[j]:
            done[j+1] = 0
            break
    done[j0] = 0

cnt = [0] * N
for i in range(N):
    cnt[par(i)] += 1
print(*[cnt[par(i)] for i in range(N)], sep = "\n")
0