結果
問題 | No.1170 Never Want to Walk |
ユーザー | あああ |
提出日時 | 2024-08-19 19:23:37 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
WA
|
実行時間 | - |
コード長 | 970 bytes |
コンパイル時間 | 412 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 35,816 KB |
最終ジャッジ日時 | 2024-08-19 19:23:48 |
合計ジャッジ時間 | 10,049 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 13 RE * 11 WA * 13 |
ソースコード
import bisectN,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 xelse: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:returnif siz[x] < siz[y]:x,y = y,xpar[x] = ysiz[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]:continuell = 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)united[i] = Trueunited[j] = Truefor j in range(rl,rr):unite(i,j)united[i] = Trueunited[j] = Truefor i in range(N):print(size(i))