結果

問題 No.1365 [Cherry 1st Tune] Whose Fault?
ユーザー 👑 KazunKazun
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

class Coloring_Union_Find():
def __init__(self,N,f,e=0):
"""0,1,...,n-1.
N:
f:2
e:
"""
self.n=N
self.parents=[-1]*N
self.data=[e]*N
self.rank=[0]*N
self.f=f
def 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]=x
return x
def union(self, x, y):
"""x,y.
x,y:
"""
x=self.find(x)
y=self.find(y)
if x==y:
return
self.data[x]=self.data[y]=self.f(self.data[x],self.data[y])
if self.rank[x]<self.rank[y]:
x,y=y,x
self.parents[x]+=self.parents[y]
self.parents[y]=x
if self.rank[x]==self.rank[y]:
self.rank[x]+=1
def 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):
"""xc.
x:
c:
"""
self.data[self.find(x)]=c
return
def 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 X
def 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 sys
input=sys.stdin.readline
write=sys.stdout.write
N=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=0
for 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]*Q
for q in range(Q):
X,Y=map(int,input().split())
if not U.same(X,Y):
d,p=U.look(X)
Ans-=d*d/p
d,p=U.look(Y)
Ans-=d*d/p
U.union(X,Y)
d,p=U.look(X)
Ans+=d*d/p
S[q]=Ans
write("\n".join(map(str,S)))
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0