結果

問題 No.2431 Viral Hotel
コンテスト
ユーザー bortik
提出日時 2023-08-14 17:07:49
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 199 ms / 2,000 ms
コード長 1,347 bytes
コンパイル時間 2,422 ms
コンパイル使用メモリ 202,192 KB
最終ジャッジ日時 2025-02-16 08:04:21
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

diff # 

//Solution to Viral Hotel problem by writer: BortiK

#include<bits/stdc++.h>

using namespace std;

int main(){
    int N, K, M, P; 
    cin >> N >> K >> M >> P;
    vector<vector<int>> adj(N);
    for(int i = 0; i < M; i++){
        int u,v; cin >> u >> v;
        u--;v--;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }

    vector<int> s(N);
    for(int i = 0; i < N; i++){
        cin >> s[i];
    }

    priority_queue<pair<int,int>, vector<pair<int,int>>, greater<pair<int,int>>> todo;
    vector<int> infected(N, -1);
    for(int i = 0; i < K; i++){
        int x; cin >> x;
        x--;
        todo.push(make_pair(s[x],x));
        infected[x] = 0;
    }

    vector<int> quar(N, -1);
    int answer = 0;
    
    while(!todo.empty()){
        pair<int,int> x = todo.top();
        todo.pop();
        if(quar[x.second] != -1 && quar[x.second] < x.first) continue;
        for(int u: adj[x.second]){
            if(infected[u] == -1){
                infected[u] = x.first;
                todo.push(make_pair(x.first+s[u], u));
            } else {
                if(infected[u]+P > x.first){
                    if(quar[u] == -1){
                        answer++;
                        quar[u] = x.first;                  
                    }
                }
            }
        }
    }

    cout << answer;
}
0