結果
問題 | No.1170 Never Want to Walk |
ユーザー |
![]() |
提出日時 | 2020-08-14 22:43:39 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,062 ms / 2,000 ms |
コード長 | 1,126 bytes |
コンパイル時間 | 318 ms |
コンパイル使用メモリ | 12,928 KB |
実行使用メモリ | 51,596 KB |
最終ジャッジ日時 | 2024-10-10 16:05:01 |
合計ジャッジ時間 | 15,542 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 37 |
ソースコード
import sys input = sys.stdin.readline import bisect N,a,b=map(int,input().split()) X=list(map(int,input().split())) # UnionFind Group = [i for i in range(N+1)] # グループ分け Nodes = [1]*(N+1) # 各グループのノードの数 def find(x): while Group[x] != x: x=Group[x] return x def Union(x,y): if find(x) != find(y): if Nodes[find(x)] < Nodes[find(y)]: Nodes[find(y)] += Nodes[find(x)] Nodes[find(x)] = 0 Group[find(x)] = find(y) else: Nodes[find(x)] += Nodes[find(y)] Nodes[find(y)] = 0 Group[find(y)] = find(x) s=t=-1 for i in range(N): s2=bisect.bisect_left(X,X[i]+a) t2=bisect.bisect_right(X,X[i]+b) if s2<t: Union(i,i-1) for j in range(t,t2): Union(i,j) else: for j in range(s2,t2): Union(i,j) s=s2 t=t2 ALIST=[[] for i in range(N)] for i in range(N): ALIST[find(i)].append(i) ANS=[-1]*N for i in range(N): for a in ALIST[i]: ANS[a]=len(ALIST[i]) for a in ANS: print(a)