結果

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

ソースコード

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 0<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