結果
問題 | No.1382 Travel in Mitaru city |
ユーザー |
👑 ![]() |
提出日時 | 2021-02-07 21:43:28 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 541 ms / 2,000 ms |
コード長 | 2,712 bytes |
コンパイル時間 | 1,022 ms |
コンパイル使用メモリ | 82,328 KB |
実行使用メモリ | 99,944 KB |
最終ジャッジ日時 | 2024-07-04 15:06:39 |
合計ジャッジ時間 | 21,782 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 68 |
ソースコード
class Union_Find():def __init__(self,N):"""0,1,...,n-1を要素として初期化する.N:要素数"""self.n=Nself.parents=[-1]*Nself.rank=[0]*Ndef find(self, x):"""要素xの属している族を調べる.x:要素"""V=[]while self.parents[x]>=0:V.append(x)x=self.parents[x]for v in V:self.parents[v]=xreturn xdef union(self, x, y):"""要素x,yを同一視する.x,y:要素"""x=self.find(x)y=self.find(y)if x==y:returnif self.rank[x]<self.rank[y]:x,y=y,xself.parents[x]+=self.parents[y]self.parents[y]=xif self.rank[x]==self.rank[y]:self.rank[x]+=1def 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が属している族の要素.※族の要素の個数が欲しいときはsizeを使うこと!!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):"""全ての族の出力"""X={r:[] for r in self.roots()}for k in range(self.n):X[self.find(k)].append(k)return Xdef refresh(self):for i in range(self.n):_=self.find(i)def __str__(self):return '\n'.join('{}: {}'.format(r, self.members(r)) for r in self.roots())def __repr__(self):return self.__str__()#================================================from collections import defaultdictimport sysinput=sys.stdin.readlineN,M,S,T=map(int,input().split())P=["*"]+list(map(int,input().split()))E=[[] for _ in range(N+1)]for _ in range(M):a,b=map(int,input().split())E[a].append(b)E[b].append(a)U=Union_Find(N+1)F=sorted(range(1,N+1),key=lambda x:P[x],reverse=True)Flag=defaultdict(int)Ans=0for x in F:for y in E[x]:if P[y]>=P[x]:U.union(x,y)if P[x]<P[S] and U.same(x,S) and Flag[P[x]]==0:Ans+=1Flag[P[x]]=1print(Ans)