結果

問題 No.2431 Viral Hotel
コンテスト
ユーザー Nachia
提出日時 2023-08-18 21:46:36
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,810 bytes
コンパイル時間 1,007 ms
コンパイル使用メモリ 88,640 KB
最終ジャッジ日時 2025-02-16 09:51:45
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 4 WA * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <queue>
#include <atcoder/modint>
using namespace std;
using i32 = int;
using u32 = unsigned int;
using i64 = long long;
using u64 = unsigned long long;
#define rep(i,n) for(int i=0; i<(int)(n); i++)
const i64 INF = 1001001001001001001;

using Modint = atcoder::static_modint<998244353>;

int main(){
    int N, K, M, P; cin >> N >> K >> M >> P;
    vector<vector<int>> adj(N);
    rep(i,M){
        int u,v; cin >> u >> v; u--; v--;
        adj[u].push_back(v);
        adj[v].push_back(u);
    }
    vector<int> S(N); rep(i,N) cin >> S[i];
    using F = pair<int, pair<int,int>>;
    priority_queue<F, vector<F>, greater<F>> que;
    vector<int> hit(N);
    vector<int> spr(N);
    vector<int> rem(N);
    vector<int> rec(N);
    int hitcount = 0;
    rep(i,K){
        int x; cin >> x; x--;
        hit[x] = 1;
        hitcount++;
        que.push({ S[x], {1,x} });
        que.push({ P, {0,x} });
    }
    while(que.size()){
        if(hitcount == 0) break;
        auto [t,oc] = que.top(); que.pop();
        auto [w,x] = oc;
        if(rem[x]) continue;
        if(w == 0){
            rec[x] = 1;
            hitcount -= hit[x]; hit[x] = 0;
        }
        else{
            for(int y : adj[x]) if(rem[y] == 0 && rec[y] == 0){
                if(hit[y]){ rem[y] = 1; hitcount--; hit[y] = 0; continue; }
                hit[y] = 1;
                hitcount++;
                que.push({ t + S[y], {1,y} });
                que.push({ t + P, {0,y} });
            }
        }
    }
    int ans = 0;
    rep(i,N) ans += rem[i];
    cout << ans << endl;
    return 0;
}


struct ios_do_not_sync{
    ios_do_not_sync(){
        ios::sync_with_stdio(false);
        cin.tie(nullptr);
    }
} ios_do_not_sync_instance;
0