結果

問題 No.3093 Safe Infection
ユーザー tnakao0123
提出日時 2025-04-08 19:23:54
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,187 bytes
コンパイル時間 870 ms
コンパイル使用メモリ 75,012 KB
実行使用メモリ 16,072 KB
最終ジャッジ日時 2025-04-08 19:24:06
合計ジャッジ時間 11,137 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 43 TLE * 1 -- * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:80:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   80 |   scanf("%d%d%d", &n, &m, &k);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~
main.cpp:81:36: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   81 |   for (int i = 0; i < n; i++) scanf("%d", as + i);
      |                               ~~~~~^~~~~~~~~~~~~~
main.cpp:85:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   85 |     scanf("%d%d", &u, &v);
      |     ~~~~~^~~~~~~~~~~~~~~~

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 3093.cc:  No.3093 Safe Infection - yukicoder
 */

#include<cstdio>
#include<vector>
#include<set>
#include<algorithm>
#include<tuple>

using namespace std;

/* constant */

const int MAX_N = 100000;
const int MAX_M = 100000;

/* typedef */

using tp4 = tuple<int,int,int,int>;
using stp4 = set<tp4>;

struct UFT {
  vector<int> links, ranks, sizes;
  UFT() {}

  void init(int n) {
    links.resize(n);
    for (int i = 0; i < n; i++) links[i] = i;
    ranks.assign(n, 1);
    sizes.assign(n, 1);
  }

  int root(int i) {
    int i0 = i;
    while (links[i0] != i0) i0 = links[i0];
    return (links[i] = i0);
  }

  int rank(int i) { return ranks[root(i)]; }
  int size(int i) { return sizes[root(i)]; }
  bool same(int i, int j) { return root(i) == root(j); }

  int merge(int i0, int i1) {
    int r0 = root(i0), r1 = root(i1), mr;
    if (r0 == r1) return r0;
    if (ranks[r0] == ranks[r1]) {
      links[r1] = r0;
      sizes[r0] += sizes[r1];
      ranks[r0]++;
      mr = r0;
    }
    else if (ranks[r0] > ranks[r1]) {
      links[r1] = r0;
      sizes[r0] += sizes[r1];
      mr = r0;
    }
    else {
      links[r0] = r1;
      sizes[r1] += sizes[r0];
      mr = r1;
    }
    return mr;
  }
};

/* global variables */

int as[MAX_N];
tp4 es[MAX_M];
UFT uft;

/* subroutines */

/* main */

int main() {
  int n, m, k;
  scanf("%d%d%d", &n, &m, &k);
  for (int i = 0; i < n; i++) scanf("%d", as + i);

  for (int i = 0; i < m; i++) {
    int u, v;
    scanf("%d%d", &u, &v);
    u--, v--;
    int a0 = as[u], a1 = as[v];
    if (a0 > a1) swap(a0, a1), swap(u, v);
    es[i] = {a0, a1, u, v};
  }
  sort(es, es + m);

  stp4 ses(es, es + m);
  uft.init(n);

  while (! ses.empty() && uft.size(0) < n) {
    auto [au, av, u, v] = *ses.begin();
    ses.erase(ses.begin());

    int ru = uft.root(u), rv = uft.root(v);
    if (ru != rv) {
      int a0 = as[ru], a1 = as[rv];
      if (a0 > a1) swap(a0, a1), swap(ru, rv);
      
      if (a0 != au || a1 != av) {
	ses.insert({a0, a1, ru, rv});
      }
      else {
	if (a1 - a0 > k) { puts("No"); return 0; }
	int r = uft.merge(ru, rv);
	as[r] = a1;
      }
    }
  }

  puts("Yes");
  return 0;
}
0