結果

問題 No.2431 Viral Hotel
コンテスト
ユーザー bortik
提出日時 2023-08-15 04:21:46
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,247 ms / 2,000 ms
コード長 829 bytes
コンパイル時間 380 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 41,984 KB
最終ジャッジ日時 2024-11-23 12:27:34
合計ジャッジ時間 26,804 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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