結果
問題 | No.1170 Never Want to Walk |
ユーザー | あああ |
提出日時 | 2024-08-19 19:35:46 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
RE
|
実行時間 | - |
コード長 | 974 bytes |
コンパイル時間 | 455 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 35,820 KB |
最終ジャッジ日時 | 2024-08-19 19:35:56 |
合計ジャッジ時間 | 8,214 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 26 RE * 11 |
ソースコード
import bisect N,A,B = map(int, input().split()) x = list(map(int, input().split())) par = [ i for i in range(N)] siz = [ 1 for i in range(N)] def root(x): if par[x] == x: return x else: par[x] = root(par[x]) return par[x] def same(x,y): return root(x) == root(y) def unite(x,y): x = root(x) y = root(y) if x == y: return if siz[x] < siz[y]: x,y = y,x par[x] = y siz[y] += siz[x] def size(x): return siz[root(x)] united = [False for i in range(N)] for i in range(N): if united[i]: continue ll = bisect.bisect_left(x,x[i]-B) lr = bisect.bisect_right(x,x[i]-A) rl = bisect.bisect_left(x,x[i]+A) rr = bisect.bisect_right(x,x[i]+B) for j in range(ll,lr): unite(i,j) if j < lr - 1: united[j] = True for j in range(rl,rr): unite(i,j) if j < rr - 1: united[j] = True for i in range(N): print(size(i))