結果
問題 | No.1170 Never Want to Walk |
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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")