結果

問題 No.2431 Viral Hotel
ユーザー sgfcsgfc
提出日時 2023-08-18 22:47:22
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 115 ms / 2,000 ms
コード長 3,974 bytes
コンパイル時間 5,208 ms
コンパイル使用メモリ 321,364 KB
実行使用メモリ 19,248 KB
最終ジャッジ日時 2024-11-28 09:00:44
合計ジャッジ時間 9,386 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 83 ms
14,900 KB
testcase_01 AC 68 ms
13,712 KB
testcase_02 AC 74 ms
14,424 KB
testcase_03 AC 82 ms
14,956 KB
testcase_04 AC 86 ms
16,500 KB
testcase_05 AC 98 ms
16,496 KB
testcase_06 AC 18 ms
7,680 KB
testcase_07 AC 17 ms
7,552 KB
testcase_08 AC 17 ms
7,552 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 2 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 115 ms
19,248 KB
testcase_13 AC 86 ms
14,628 KB
testcase_14 AC 43 ms
9,644 KB
testcase_15 AC 17 ms
5,996 KB
testcase_16 AC 56 ms
14,068 KB
testcase_17 AC 34 ms
8,604 KB
testcase_18 AC 87 ms
17,284 KB
testcase_19 AC 55 ms
11,856 KB
testcase_20 AC 23 ms
7,680 KB
testcase_21 AC 21 ms
6,988 KB
testcase_22 AC 54 ms
12,172 KB
testcase_23 AC 56 ms
11,616 KB
testcase_24 AC 32 ms
8,636 KB
testcase_25 AC 73 ms
13,468 KB
testcase_26 AC 44 ms
10,832 KB
testcase_27 AC 49 ms
10,572 KB
testcase_28 AC 40 ms
9,508 KB
testcase_29 AC 40 ms
9,744 KB
testcase_30 AC 47 ms
10,440 KB
testcase_31 AC 66 ms
13,584 KB
testcase_32 AC 23 ms
6,380 KB
testcase_33 AC 37 ms
9,440 KB
testcase_34 AC 6 ms
5,248 KB
testcase_35 AC 13 ms
5,504 KB
testcase_36 AC 76 ms
13,288 KB
testcase_37 AC 52 ms
11,164 KB
testcase_38 AC 22 ms
7,552 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 2 ms
5,248 KB
testcase_43 AC 101 ms
18,220 KB
testcase_44 AC 65 ms
12,264 KB
testcase_45 AC 48 ms
9,800 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#include <unordered_map>
#include <unordered_set>

#include <stdio.h>
#include <math.h>
#include <cassert>

#pragma warning(push)
#pragma warning(disable:26498 26450 26451 26812)
#include "atcoder/all"
#pragma warning(pop)

#include <float.h>


//#include <boost/multiprecision/cpp_int.hpp>
//using namespace boost::multiprecision;
// std::fixed << std::setprecision(10) <<

using namespace std;
using namespace atcoder;

using ll = long long;
using ull = unsigned long long;
using mint = modint998244353;
using mint2 = modint1000000007;

const double eps = 1e-9;

#define REP(i, n) for (ll i = 0; i < ll(n); ++i)
#define REPI(i, init, n) for (ll i = init; i < ll(n); ++i)
#define REPD(i, init) for (ll i = init; i >=0; --i)
#define REPDI(i, init, n) for (ll i = init; i >=n; --i)
#define REPE(i, c) for (const auto& i : c)

using vl = vector<ll>;
#define VL(a, n) vector<ll> a; a.assign(n, 0);
#define VLI(a, n) vector<ll> a; a.assign(n, 0); for(auto& x : a) cin >> x;
#define VSI(a, n) vector<string> a; a.assign(n, ""); for(auto& x : a) cin >> x;
using vvl = vector<vector<ll>>;
#define VVLI(a, n, m, init) vector<vector<ll>> a(n); for(auto& x : a) x.assign(m, init);
using pl = pair<ll, ll>;

struct uv { ll u; ll v; ll c; };
#define VUVI(a, n) vector<uv> a; a.assign(n, {0, 0, 1}); for(auto& x : a) {cin >> x.u >> x.v; x.u--; x.v--;}
#define VUVCI(a, n) vector<uv> a; a.assign(n, {0, 0, 0}); for(auto& x : a) {cin >> x.u >> x.v >> x.c; x.u--; x.v--;}
vvl to_edge(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) ret[x.u].push_back(x.v); return ret; }
vvl to_edge_d(const ll n, const vector<uv>& v) { vvl ret(n); for (auto& x : v) { ret[x.u].push_back(x.v); ret[x.v].push_back(x.u); } return ret; }

template <class T = long long> using pql = priority_queue<T>;
template <class T = long long> using pqg = priority_queue<T, vector<T>, greater<T>>;
using vm = vector<mint>;
using vvm = vector<vm>;

const char* YES = "Yes";
const char* NO = "No";
void yn(bool f) { std::cout << (f ? YES : NO) << endl; };
template<class T> void ov(const T& v) { for (auto it = v.begin(); it != v.end(); it++) { if (it != v.begin()) cout << " "; cout << *it; } };
template<> void ov(const vm& v) { for (auto it = v.begin(); it != v.end(); it++) { if (it != v.begin()) cout << " "; cout << it->val(); } };

const ll llhuge = 1LL << 60; //十分でかいが多少足しても溢れない数値
template<class T, class U> bool chmin(T& a, const U& b) { a = min(a, b); return b == a; }
template<class T, class U> bool chmax(T& a, const U& b) { a = max(a, b); return b == a; }
template<class T> void mysort(T& v) { std::sort(begin(v), end(v)); };
template<class T, class U> void mysort(T& v, U pr) { std::sort(begin(v), end(v), pr); };
template<class T> void myrev(T& v) { std::reverse(begin(v), end(v)); };


int main() {
	cin.tie(nullptr);
	ios::sync_with_stdio(false);

	ll n, k, m, p;
	cin >> n >> k >> m >> p;

	VUVI(uv, m);
	auto g = to_edge_d(n, uv);

	vl st(n, 0); //0未感染 1感染 2免疫 3不在
	VLI(s, n);

	pqg<pair<ll, pair<ll, ll>>> q;

	REP(i, k) {
		ll x;
		cin >> x;
		x--;
		st[x] = 1;
		q.push({ s[x], {2, x} }); //感染イベント
		q.push({ p, {1, x} }); //回復イベント
	}

	ll ans = 0;
	while (!q.empty()) {
		auto t = q.top();
		q.pop();
		ll now = t.first;

		if (t.second.first == 0) {
			//検疫
			if (st[t.second.second] == 1) {
				st[t.second.second] = 3;
				ans++;
			}
		}
		if (t.second.first == 1) {
			//回復
			if (st[t.second.second] == 1) {
				st[t.second.second] = 2;
			}
		}
		if (t.second.first == 2 && st[t.second.second] < 3) {
			//感染
			REPE(e, g[t.second.second]) {
				if (st[e] == 0) {
					st[e] = 1;
					q.push({ now + s[e], {2, e} }); //感染イベント
					q.push({ now + p, {1, e} }); //回復イベント
				}
				else if (st[e] == 1) {
					q.push({ now + 1, {0, e} }); //検疫イベント //明日一番にやる
				}
			}
		}
	}

	cout << ans << endl;

	return 0;
}
0