結果

問題 No.3093 Safe Infection
コンテスト
ユーザー ゼット
提出日時 2025-04-06 16:40:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 258 ms / 2,000 ms
コード長 1,137 bytes
コンパイル時間 353 ms
コンパイル使用メモリ 82,656 KB
実行使用メモリ 97,428 KB
最終ジャッジ日時 2025-04-06 16:40:53
合計ジャッジ時間 12,839 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 70
権限があれば一括ダウンロードができます

ソースコード

diff # 

class unif:
  def __init__(self,n):
    self.pare=[-1]*n
    self.size=[1]*n
    self.big=[0]*n
  def root(self,x):
    while self.pare[x]!=-1:
      x=self.pare[x]
    return x
  def unite(self,u,v):
    rootu=self.root(u)
    rootv=self.root(v)
    if rootu!=rootv:
      if self.size[rootu]>=self.size[rootv]:
        self.pare[rootv]=rootu
        self.size[rootu]+=self.size[rootv]
        self.big[rootu]=max(self.big[rootu],self.big[rootv])
      else:
        self.pare[rootu]=rootv
        self.size[rootv]+=self.size[rootu]
        self.big[rootv]=max(self.big[rootu],self.big[rootv])
  def same(self,s,t):
    return self.root(s)==self.root(t)
N,M,K=map(int,input().split())
A=list(map(int,input().split()))
G=[[] for i in range(N)]
for i in range(M):
  a,b=map(int,input().split())
  a-=1
  b-=1
  if A[a]>A[b]:
    G[a].append(b)
  else:
    G[b].append(a)
L=[]
for i in range(N):
  L.append((A[i],i))
L.sort()
Z=unif(N)
for i in range(N):
  Z.big[i]=A[i]
for i in range(N):
  a=L[i][1]
  for b in G[a]:
    x=Z.root(a)
    y=Z.root(b)
    if Z.big[x]>Z.big[y]+K:
      print('No')
      exit()
    Z.unite(a,b)
print('Yes')
0