結果

問題 No.2431 Viral Hotel
ユーザー pockyny
提出日時 2023-08-18 23:23:19
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,382 bytes
コンパイル時間 867 ms
コンパイル使用メモリ 84,400 KB
最終ジャッジ日時 2025-02-16 10:49:14
ジャッジサーバーID
(参考情報)
judge5 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 4 WA * 38
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <iostream>
#include <set>
#include <vector>

using namespace std;
typedef pair<int,int> pii;
vector<int> G[100010];
int s[100010],c[100010];
bool cure[100010],iru[100010];
int main(){
    int i,n,k,m,p; cin >> n >> k >> m >> p;
    for(i=0;i<m;i++){
        int u,v; cin >> u >> v; u--; v--;
        G[u].push_back(v); G[v].push_back(u);
    }
    for(i=0;i<n;i++){
        cure[i] = false; iru[i] = true;
    }
    set<pair<pii,int>> ms;
    for(i=0;i<n;i++) cin >> s[i];
    for(i=0;i<k;i++){
        int x; cin >> x; x--;
        ms.insert({{p,-1},x});
        ms.insert({{s[x],1},x});
        c[x] = 1;
    }
    while(ms.size()){
        auto x = *ms.begin();
        ms.erase(ms.find(x));
        if(x.first.second==-1){
            cure[x.second] = true;
        }else{
            if(!iru[x.second]) continue;
            for(int u:G[x.second]){
                if(!cure[u]){
                    if(!c[u]){
                        c[u] = 1;
                        ms.insert({{x.first.first + p,-1},u});
                        ms.insert({{x.first.first + s[u],1},u});
                    }else{
                        iru[u] = false;
                    }
                }
            }
        }
    }
    int ans = 0;
    for(i=0;i<n;i++){
        if(!iru[i]) ans++;
    }
    cout << ans << "\n";
    // for(i=0;i<n;i++) cout << iru[i];
    // cout << endl;
}
0