結果

問題 No.2431 Viral Hotel
ユーザー 👑 NachiaNachia
提出日時 2023-08-18 22:09:01
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 131 ms / 2,000 ms
コード長 1,914 bytes
コンパイル時間 1,016 ms
コンパイル使用メモリ 88,176 KB
最終ジャッジ日時 2025-02-16 10:05:48
ジャッジサーバーID
(参考情報)
judge5 / 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;
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0