結果
問題 | No.1170 Never Want to Walk |
ユーザー |
👑 ![]() |
提出日時 | 2020-08-14 21:29:40 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 2,084 bytes |
コンパイル時間 | 263 ms |
コンパイル使用メモリ | 82,304 KB |
実行使用メモリ | 106,752 KB |
最終ジャッジ日時 | 2024-10-10 14:10:05 |
合計ジャッジ時間 | 6,470 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 25 TLE * 1 -- * 11 |
ソースコード
class Union_Find():def __init__(self,N):"""0,1,...,n-1を要素として初期化する.n:要素数"""self.n=Nself.parents=[-1]*Ndef find(self, x):"""要素xの属している族を調べる.x:要素"""if self.parents[x] < 0:return xelse:self.parents[x] = self.find(self.parents[x])return self.parents[x]def union(self, x, y):"""要素x,yを同一視する.x,y:要素"""x = self.find(x)y = self.find(y)if x == y:returnif self.parents[x] > self.parents[y]:x, y = y, xself.parents[x] += self.parents[y]self.parents[y] = xdef size(self, x):"""要素xの属している要素の数.x:要素"""return -self.parents[self.find(x)]def same(self, x, y):"""要素x,yは同一視されているか?x,y:要素"""return self.find(x) == self.find(y)def members(self, x):"""要素xが属している族の要素.x:要素"""root = self.find(x)return [i for i in range(self.n) if self.find(i) == root]def roots(self):"""族の名前のリスト"""return [i for i, x in enumerate(self.parents) if x < 0]def group_count(self):"""族の個数"""return len(self.roots())def all_group_members(self):"""全ての族の出力"""return {r: self.members(r) for r in self.roots()}def __str__(self):return '\n'.join('{}: {}'.format(r, self.members(r)) for r in self.roots())#================================================N,A,B=map(int,input().split())X=list(map(int,input().split()))U=Union_Find(N)for i in range(N):for j in range(i+1,N):if A<=abs(X[i]-X[j])<=B:U.union(i,j)Ans=[0]*Nfor i in range(N):Ans[i]=len(U.members(i))print("\n".join(map(str,Ans)))