結果
問題 | No.1365 [Cherry 1st Tune] Whose Fault? |
ユーザー |
👑 ![]() |
提出日時 | 2021-01-17 02:33:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 613 ms / 2,000 ms |
コード長 | 3,469 bytes |
コンパイル時間 | 308 ms |
コンパイル使用メモリ | 82,704 KB |
実行使用メモリ | 116,128 KB |
最終ジャッジ日時 | 2024-12-29 09:22:15 |
合計ジャッジ時間 | 15,535 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 46 |
ソースコード
class Coloring_Union_Find():def __init__(self,N,f,e=0):"""0,1,...,n-1を要素として初期化する.N:要素数f:2変数関数の合成e:最初の値"""self.n=Nself.parents=[-1]*Nself.data=[e]*Nself.rank=[0]*Nself.f=fdef 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:returnself.data[x]=self.data[y]=self.f(self.data[x],self.data[y])if 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 set(self,x,c):"""要素xの属する成分の色をcに変更する.x:要素c:色"""self.data[self.find(x)]=creturndef look(self,x):"""要素xの属する成分の色x:要素"""return self.data[self.find(x)]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 color_list(self):return [self.look(x) for x in range(self.n)]def color_map(self):return {x:self.look(x) for x in self.roots()}def __str__(self):return '\n'.join('{} [color:{}]: {}'.format(r,self.look(r),self.members(r)) for r in self.roots())def __repr__(self):return self.__str__()#================================================import sysinput=sys.stdin.readlinewrite=sys.stdout.writeN=int(input())A=["*"]+list(map(int,input().split()))B=["*"]+list(map(int,input().split()))P=["*"]+list(map(int,input().split()))inf=float("inf")U=Coloring_Union_Find(N+1,lambda x,y:[x[0]+y[0],x[1]+y[1]],[0,0])Ans=0for i in range(1,N+1):d=A[i]-B[i]Ans+=d*d*P[i]if P[i]:U.set(i,[d,1/P[i]])else:U.set(i,[d,inf])Q=int(input())S=[0]*Qfor q in range(Q):X,Y=map(int,input().split())if not U.same(X,Y):d,p=U.look(X)Ans-=d*d/pd,p=U.look(Y)Ans-=d*d/pU.union(X,Y)d,p=U.look(X)Ans+=d*d/pS[q]=Answrite("\n".join(map(str,S)))