結果

問題 No.2431 Viral Hotel
ユーザー 👑 NachiaNachia
提出日時 2023-08-18 22:09:01
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 118 ms / 2,000 ms
コード長 1,914 bytes
コンパイル時間 933 ms
コンパイル使用メモリ 91,880 KB
実行使用メモリ 13,464 KB
最終ジャッジ日時 2024-05-06 04:13:34
合計ジャッジ時間 4,894 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 85 ms
10,696 KB
testcase_01 AC 72 ms
10,100 KB
testcase_02 AC 75 ms
10,708 KB
testcase_03 AC 91 ms
11,520 KB
testcase_04 AC 89 ms
12,380 KB
testcase_05 AC 104 ms
12,380 KB
testcase_06 AC 17 ms
5,376 KB
testcase_07 AC 16 ms
5,376 KB
testcase_08 AC 15 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 2 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 118 ms
13,464 KB
testcase_13 AC 88 ms
11,220 KB
testcase_14 AC 42 ms
7,936 KB
testcase_15 AC 18 ms
5,376 KB
testcase_16 AC 53 ms
10,716 KB
testcase_17 AC 37 ms
6,764 KB
testcase_18 AC 83 ms
12,004 KB
testcase_19 AC 55 ms
9,616 KB
testcase_20 AC 25 ms
6,656 KB
testcase_21 AC 20 ms
5,616 KB
testcase_22 AC 55 ms
8,604 KB
testcase_23 AC 59 ms
8,668 KB
testcase_24 AC 35 ms
7,680 KB
testcase_25 AC 77 ms
10,324 KB
testcase_26 AC 48 ms
9,088 KB
testcase_27 AC 51 ms
8,396 KB
testcase_28 AC 42 ms
7,548 KB
testcase_29 AC 45 ms
7,168 KB
testcase_30 AC 49 ms
7,512 KB
testcase_31 AC 68 ms
10,420 KB
testcase_32 AC 19 ms
5,376 KB
testcase_33 AC 38 ms
6,944 KB
testcase_34 AC 5 ms
5,376 KB
testcase_35 AC 14 ms
5,376 KB
testcase_36 AC 77 ms
10,632 KB
testcase_37 AC 50 ms
8,380 KB
testcase_38 AC 23 ms
6,912 KB
testcase_39 AC 2 ms
5,376 KB
testcase_40 AC 2 ms
5,376 KB
testcase_41 AC 1 ms
5,376 KB
testcase_42 AC 2 ms
5,376 KB
testcase_43 AC 100 ms
13,328 KB
testcase_44 AC 63 ms
9,396 KB
testcase_45 AC 47 ms
7,936 KB
権限があれば一括ダウンロードができます

ソースコード

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