結果

問題 No.2431 Viral Hotel
ユーザー 👑 NachiaNachia
提出日時 2023-08-18 22:09:01
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 140 ms / 2,000 ms
コード長 1,914 bytes
コンパイル時間 1,251 ms
コンパイル使用メモリ 91,544 KB
実行使用メモリ 13,456 KB
最終ジャッジ日時 2024-11-28 07:30:02
合計ジャッジ時間 6,424 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 42
権限があれば一括ダウンロードができます

ソースコード

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 if(w == 1){
            for(int y : adj[x]) if(rem[y] == 0 && rec[y] == 0){
                if(hit[y]){ que.push({ t, {2,y} }); continue; }
                hit[y] = 1;
                hitcount++;
                que.push({ t + S[y], {1,y} });
                que.push({ t + P, {0,y} });
            }
        }
        else{
            rem[x] = 1;
            hitcount -= hit[x];
            hit[x] = 0;
        }
    }
    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