結果

問題 No.2431 Viral Hotel
コンテスト
ユーザー 0214sh7
提出日時 2023-08-14 20:30:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,699 ms / 2,000 ms
コード長 1,489 bytes
コンパイル時間 260 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 115,100 KB
最終ジャッジ日時 2024-11-23 12:26:24
合計ジャッジ時間 35,168 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

import heapq

N,K,M,P = map(int,input().split())
G = [ [] for _ in range(N)]
for i in range(M):
    u,v = map(int,input().split())
    u -= 1;v -= 1
    G[u].append(v)
    G[v].append(u)
S = [int(input()) for _ in range(N)]
X = [int(input())-1 for _ in range(K)]

# {イベントが起こる日,イベントの種類,起こる頂点}
que = []
heapq.heapify(que)
Ans = 0

#回復したか
recovered = [False]*N
#検疫が入ったか
quarantined = [False]*N
#いま感染しているか
infected = [False]*N

#0: 回復 1: 隣に感染 2:検疫 
for i in range(K):
    infected[X[i]] = True
    heapq.heappush(que, [P,0,X[i]])
    heapq.heappush(que, [S[X[i]],1,X[i]])

while len(que)>0:
    q = heapq.heappop(que)

    #検疫が入った場合その後一切のアクションを起こさない
    if quarantined[q[2]]:
        continue

    if q[1] == 0:
        #回復
        infected[q[2]] = False
        recovered[q[2]] = True
    elif q[1] == 1:
        #隣にうつす
        for e in G[q[2]]:
            if recovered[e] or quarantined[e]:
                continue
            #すでにかかっていた場合検疫送り
            if infected[e]:
                heapq.heappush(que, [q[0],2,e])
            else:
                infected[e] = True
                heapq.heappush(que, [q[0]+P,0,e])
                heapq.heappush(que, [q[0]+S[e],1,e])
    else:
        #検疫
        Ans += 1
        infected[q[2]] = False
        quarantined[q[2]] = True

print(Ans)
0