結果

問題 No.3093 Safe Infection
ユーザー sasa8uyauya
提出日時 2025-04-11 17:11:12
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 654 bytes
コンパイル時間 514 ms
コンパイル使用メモリ 82,368 KB
実行使用メモリ 105,816 KB
最終ジャッジ日時 2025-04-11 17:11:39
合計ジャッジ時間 21,497 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other WA * 70
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

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]

for s in sorted(list(range(n)),key=lambda i:a[i]):
  e[s].sort(key=lambda i:a[i])
  for t in e[s]:
    if abs(a[root(t)]-a[root(s)])<=k:
      union(s,t)

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

print(r)
0