結果

問題 No.2431 Viral Hotel
ユーザー 👑 NachiaNachia
提出日時 2023-08-18 22:09:01
言語 C++17
(gcc 12.3.0 + boost 1.83.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 102 ms
10,820 KB
testcase_01 AC 87 ms
10,224 KB
testcase_02 AC 90 ms
10,580 KB
testcase_03 AC 115 ms
11,648 KB
testcase_04 AC 116 ms
12,504 KB
testcase_05 AC 129 ms
12,508 KB
testcase_06 AC 19 ms
5,248 KB
testcase_07 AC 18 ms
5,248 KB
testcase_08 AC 18 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 3 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 140 ms
13,432 KB
testcase_13 AC 115 ms
11,092 KB
testcase_14 AC 53 ms
7,936 KB
testcase_15 AC 18 ms
5,248 KB
testcase_16 AC 65 ms
10,584 KB
testcase_17 AC 43 ms
6,632 KB
testcase_18 AC 105 ms
12,136 KB
testcase_19 AC 70 ms
9,492 KB
testcase_20 AC 33 ms
6,784 KB
testcase_21 AC 26 ms
5,488 KB
testcase_22 AC 72 ms
8,740 KB
testcase_23 AC 79 ms
8,540 KB
testcase_24 AC 42 ms
7,680 KB
testcase_25 AC 103 ms
10,324 KB
testcase_26 AC 62 ms
8,960 KB
testcase_27 AC 71 ms
8,528 KB
testcase_28 AC 55 ms
7,548 KB
testcase_29 AC 60 ms
7,296 KB
testcase_30 AC 68 ms
7,640 KB
testcase_31 AC 93 ms
10,548 KB
testcase_32 AC 27 ms
5,248 KB
testcase_33 AC 56 ms
6,872 KB
testcase_34 AC 6 ms
5,248 KB
testcase_35 AC 18 ms
5,248 KB
testcase_36 AC 101 ms
10,756 KB
testcase_37 AC 75 ms
8,376 KB
testcase_38 AC 33 ms
6,784 KB
testcase_39 AC 2 ms
5,248 KB
testcase_40 AC 2 ms
5,248 KB
testcase_41 AC 2 ms
5,248 KB
testcase_42 AC 3 ms
5,248 KB
testcase_43 AC 139 ms
13,456 KB
testcase_44 AC 79 ms
9,392 KB
testcase_45 AC 67 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