結果

問題 No.3093 Safe Infection
ユーザー sasa8uyauya
提出日時 2025-04-11 16:49:03
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 579 bytes
コンパイル時間 284 ms
コンパイル使用メモリ 82,360 KB
実行使用メモリ 104,812 KB
最終ジャッジ日時 2025-04-11 16:49:20
合計ジャッジ時間 15,422 ms
ジャッジサーバーID
(参考情報)
judge5 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 16 WA * 54
権限があれば一括ダウンロードができます

ソースコード

diff #

n,m,k=map(int,input().split())
a=list(map(int,input().split()))
e=[]
for _ in range(m):
  u,v=map(int,input().split())
  u-=1
  v-=1
  if a[u]>a[v]:
    u,v=v,u
  e+=[(a[u],a[v]-a[u],u,v)]

r=list(range(n))

def union(x,y):
  rx=root(x)
  ry=root(y)
  if rx==ry:
    return
  if a[rx]<a[ry]:
    rx,ry=ry,rx
  r[ry]=rx
  return

def root(x):
  p=x
  l=[p]
  while r[p]!=p:
    p=r[p]
    l+=[p]
  for p in l:
    r[p]=l[-1]
  return r[x]
e.sort()
for _,_,u,v in e:
  if abs(a[root(v)]-a[root(u)])<=k:
    union(u,v)

print(["No","Yes"][len(set(a[root(i)] for i in range(n)))==1])
0