結果

問題 No.2431 Viral Hotel
ユーザー tnakao0123tnakao0123
提出日時 2023-08-22 16:06:00
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 87 ms / 2,000 ms
コード長 1,722 bytes
コンパイル時間 702 ms
コンパイル使用メモリ 61,552 KB
実行使用メモリ 12,544 KB
最終ジャッジ日時 2024-05-09 21:52:07
合計ジャッジ時間 5,751 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 65 ms
10,196 KB
testcase_01 AC 55 ms
9,712 KB
testcase_02 AC 58 ms
10,116 KB
testcase_03 AC 65 ms
10,112 KB
testcase_04 AC 72 ms
10,888 KB
testcase_05 AC 80 ms
11,164 KB
testcase_06 AC 19 ms
6,400 KB
testcase_07 AC 20 ms
6,528 KB
testcase_08 AC 20 ms
6,400 KB
testcase_09 AC 3 ms
5,376 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 87 ms
12,392 KB
testcase_13 AC 69 ms
9,612 KB
testcase_14 AC 34 ms
7,936 KB
testcase_15 AC 15 ms
6,284 KB
testcase_16 AC 47 ms
10,096 KB
testcase_17 AC 31 ms
7,496 KB
testcase_18 AC 67 ms
11,548 KB
testcase_19 AC 43 ms
9,012 KB
testcase_20 AC 21 ms
7,040 KB
testcase_21 AC 18 ms
6,764 KB
testcase_22 AC 42 ms
8,884 KB
testcase_23 AC 45 ms
8,448 KB
testcase_24 AC 28 ms
7,624 KB
testcase_25 AC 62 ms
9,220 KB
testcase_26 AC 39 ms
8,276 KB
testcase_27 AC 43 ms
8,368 KB
testcase_28 AC 35 ms
7,808 KB
testcase_29 AC 35 ms
7,728 KB
testcase_30 AC 40 ms
8,144 KB
testcase_31 AC 52 ms
9,764 KB
testcase_32 AC 18 ms
6,528 KB
testcase_33 AC 33 ms
7,544 KB
testcase_34 AC 6 ms
5,760 KB
testcase_35 AC 13 ms
6,400 KB
testcase_36 AC 59 ms
10,068 KB
testcase_37 AC 41 ms
8,840 KB
testcase_38 AC 22 ms
7,168 KB
testcase_39 AC 3 ms
5,504 KB
testcase_40 AC 3 ms
5,376 KB
testcase_41 AC 3 ms
5,376 KB
testcase_42 AC 2 ms
5,504 KB
testcase_43 AC 73 ms
12,544 KB
testcase_44 AC 48 ms
9,484 KB
testcase_45 AC 37 ms
8,064 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