結果

問題 No.2431 Viral Hotel
ユーザー bortik
提出日時 2023-08-15 04:22:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 596 ms / 2,000 ms
コード長 829 bytes
コンパイル時間 440 ms
コンパイル使用メモリ 82,276 KB
実行使用メモリ 96,568 KB
最終ジャッジ日時 2024-11-23 12:27:04
合計ジャッジ時間 17,416 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq

N,K,M,P = map(int,input().split())

adj = [ [] for i in range(N)]

for i in range(M):
    u,v = map(int, input().split())
    u -= 1
    v -= 1
    adj[u].append(v)
    adj[v].append(u)

s = [int(input()) for _ in range(N)]

infected = [-1 for _ in range(N)]

todo = []
heapq.heapify(todo)

for i in range(K):
    x = int(input())
    x -= 1
    infected[x] = 0
    heapq.heappush(todo, (s[x], x))

quar = [-1 for _ in range(N)]
answer = 0

while todo:
    x = heapq.heappop(todo)
    if quar[x[1]] != -1 and quar[x[1]] < x[0]: continue
    for u in adj[x[1]]:
        if infected[u] == -1:
            infected[u] = x[0]
            heapq.heappush(todo, (x[0]+s[u], u))
        else:
            if infected[u] + P > x[0] and quar[u] == -1:
                quar[u] = x[0]
                answer += 1

print(answer)
0