結果

問題 No.2431 Viral Hotel
ユーザー tnakao0123tnakao0123
提出日時 2023-08-22 16:06:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 91 ms / 2,000 ms
コード長 1,722 bytes
コンパイル時間 804 ms
コンパイル使用メモリ 60,976 KB
実行使用メモリ 12,096 KB
最終ジャッジ日時 2023-08-22 16:06:12
合計ジャッジ時間 5,600 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 73 ms
9,944 KB
testcase_01 AC 61 ms
9,616 KB
testcase_02 AC 66 ms
9,784 KB
testcase_03 AC 72 ms
9,996 KB
testcase_04 AC 78 ms
10,744 KB
testcase_05 AC 85 ms
11,016 KB
testcase_06 AC 19 ms
6,300 KB
testcase_07 AC 20 ms
6,352 KB
testcase_08 AC 20 ms
6,348 KB
testcase_09 AC 3 ms
5,332 KB
testcase_10 AC 3 ms
5,396 KB
testcase_11 AC 3 ms
5,336 KB
testcase_12 AC 91 ms
11,972 KB
testcase_13 AC 77 ms
9,576 KB
testcase_14 AC 38 ms
7,820 KB
testcase_15 AC 16 ms
6,156 KB
testcase_16 AC 47 ms
9,804 KB
testcase_17 AC 31 ms
7,360 KB
testcase_18 AC 77 ms
11,240 KB
testcase_19 AC 47 ms
8,924 KB
testcase_20 AC 23 ms
6,968 KB
testcase_21 AC 19 ms
6,752 KB
testcase_22 AC 47 ms
8,752 KB
testcase_23 AC 53 ms
8,312 KB
testcase_24 AC 32 ms
7,372 KB
testcase_25 AC 68 ms
9,136 KB
testcase_26 AC 41 ms
8,264 KB
testcase_27 AC 49 ms
8,092 KB
testcase_28 AC 39 ms
7,712 KB
testcase_29 AC 38 ms
7,656 KB
testcase_30 AC 44 ms
7,800 KB
testcase_31 AC 58 ms
9,512 KB
testcase_32 AC 19 ms
6,292 KB
testcase_33 AC 40 ms
7,596 KB
testcase_34 AC 6 ms
5,644 KB
testcase_35 AC 14 ms
6,180 KB
testcase_36 AC 63 ms
9,604 KB
testcase_37 AC 43 ms
8,516 KB
testcase_38 AC 23 ms
6,940 KB
testcase_39 AC 3 ms
5,340 KB
testcase_40 AC 3 ms
5,332 KB
testcase_41 AC 3 ms
5,396 KB
testcase_42 AC 3 ms
5,368 KB
testcase_43 AC 80 ms
12,096 KB
testcase_44 AC 50 ms
9,224 KB
testcase_45 AC 41 ms
7,908 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 2431.cc:  No.2431 Viral Hotel - yukicoder
 */

#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
 
using namespace std;

/* constant */

const int MAX_N = 100000;

enum { NEG, POS, QRT, IMM };

/* typedef */

typedef vector<int> vi;

struct Event {
  int ev, t, u;
  Event() {}
  Event(int _ev, int _t, int _u): ev(_ev), t(_t), u(_u) {}

  bool operator<(const Event &e) const { return t > e.t; }
};

/* global variables */

vi nbrs[MAX_N];
int ss[MAX_N], xs[MAX_N], sts[MAX_N];

/* subroutines */

/* main */

int main() {
  int n, k, m, p;
  scanf("%d%d%d%d", &n, &k, &m, &p);

  for (int i = 0; i < m; i++) {
    int u, v;
    scanf("%d%d", &u, &v);
    u--, v--;
    nbrs[u].push_back(v);
    nbrs[v].push_back(u);
  }

  for (int i = 0; i < n; i++) scanf("%d", ss + i);
  for (int i = 0; i < k; i++) scanf("%d", xs + i), xs[i]--;

  fill(sts, sts + n, NEG);
  priority_queue<Event> q;
  for (int i = 0; i < k; i++) {
    sts[xs[i]] = POS;
    q.push(Event(POS, ss[xs[i]], xs[i]));
    q.push(Event(IMM, p, xs[i]));
  }

  while (! q.empty()) {
    vi imv, psv;
    int t = q.top().t;
    while (! q.empty() && q.top().t == t) {
      auto e = q.top(); q.pop();
      if (sts[e.u] != QRT) {
	if (e.ev == IMM) imv.push_back(e.u);
	else if (e.ev == POS) psv.push_back(e.u);
      }
    }

    for (auto u: imv)
      sts[u] = IMM;

    for (auto u: psv)
      for (auto v: nbrs[u]) {
	if (sts[v] == NEG) {
	  sts[v] = POS;
	  q.push(Event(POS, t + ss[v], v));
	  q.push(Event(IMM, t + p, v));
	}
	else if (sts[v] == POS) {
	  sts[v] = QRT;
	}
      }
  }

  int cnt = 0;
  for (int u = 0; u < n; u++)
    if (sts[u] == QRT) cnt++;

  printf("%d\n", cnt);
  return 0;
}
0