結果

問題 No.2431 Viral Hotel
ユーザー Manuel1024
提出日時 2023-08-18 23:56:23
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,153 bytes
コンパイル時間 1,228 ms
コンパイル使用メモリ 101,388 KB
実行使用メモリ 12,220 KB
最終ジャッジ日時 2024-11-28 11:43:17
合計ジャッジ時間 6,636 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 4 WA * 38
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
using namespace std;
using ll = long long;
using P = pair<int, int>;
constexpr ll MOD = 998244353;

int main(){
    int n, m, k, p; cin >> n >> k >> m >> p;
    vector<vector<int>> G(n);
    for(int i = 0; i < m; i++){
        int u, v; cin >> u >> v;
        u--; v--;
        G[u].emplace_back(v);
        G[v].emplace_back(u);
    }
    vector<int> s(n);
    for(auto &it: s) cin >> it;
    vector<int> x(k);
    for(auto &it: x){
        cin >> it;
        it--;
    }

    vector<map<int, vector<int>>> events(2);
    vector<int> states(n, 0);
    vector<int> cnt(n, 0);

    priority_queue<int, vector<int>, greater<int>> days;
    for(const auto &it: x){
        events[1][s[it]].emplace_back(it);
        days.emplace(s[it]);
        events[0][p].emplace_back(it);
        days.emplace(p);
        cnt[it]++;
        states[it] = 1;
    }
    set<int> past;
    while(!days.empty()){
        const int day = days.top(); days.pop();
        if(past.count(day) == 1) continue;
        past.emplace(day);
        if(events[0].count(day) == 1){
            for(const auto &it: events[0][day]){
                if(states[it] == 3) continue;
                states[it] = 2;
            }
        }
        if(events[1].count(day) == 1){
            for(const auto &it: events[1][day]){
                if(states[it] == 3) continue;
                for(const auto &nex: G[it]){
                    if(2 <= states[nex]) continue;
                    cnt[nex]++;
                    if(cnt[nex] == 1){
                        events[1][day+s[nex]].emplace_back(nex);
                        days.emplace(day+s[nex]);
                        events[0][day+p].emplace_back(nex);
                        days.emplace(day+p);
                        states[nex] = 1;
                    }else{
                        states[nex] = 3;
                    }
                }
            }
        }
    }
    int ans = 0;
    for(int i = 0; i < n; i++){
        if(states[i] == 3) ans++;
    }
    cout << ans << endl;
    return 0;
}
0