結果

問題 No.2642 Don't cut line!
コンテスト
ユーザー Misuki
提出日時 2024-02-19 22:29:27
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 122 ms / 4,000 ms
コード長 3,936 bytes
コンパイル時間 3,025 ms
コンパイル使用メモリ 229,912 KB
実行使用メモリ 14,724 KB
最終ジャッジ日時 2024-09-29 03:35:40
合計ジャッジ時間 7,103 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O2")
#include <algorithm>
#include <array>
#include <bit>
#include <bitset>
#include <cassert>
#include <cctype>
#include <cfenv>
#include <cfloat>
#include <chrono>
#include <cinttypes>
#include <climits>
#include <cmath>
#include <compare>
#include <complex>
#include <concepts>
#include <cstdarg>
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <fstream>
#include <functional>
#include <initializer_list>
#include <iomanip>
#include <ios>
#include <iostream>
#include <istream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <memory>
#include <new>
#include <numbers>
#include <numeric>
#include <ostream>
#include <queue>
#include <random>
#include <ranges>
#include <set>
#include <span>
#include <sstream>
#include <stack>
#include <streambuf>
#include <string>
#include <tuple>
#include <type_traits>
#include <variant>

#define int ll
#define INT128_MAX (__int128)(((unsigned __int128) 1 << ((sizeof(__int128) * __CHAR_BIT__) - 1)) - 1)
#define INT128_MIN (-INT128_MAX - 1)

#define clock chrono::steady_clock::now().time_since_epoch().count()

#ifdef DEBUG
#define dbg(x) cout << (#x) << " = " << x << '\n'
#else
#define dbg(x)
#endif

namespace R = std::ranges;
namespace V = std::views;

using namespace std;

using ll = long long;
using ull = unsigned long long;
using ldb = long double;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
//#define double ldb

template<class T>
ostream& operator<<(ostream& os, const pair<T, T> pr) {
  return os << pr.first << ' ' << pr.second;
}
template<class T, size_t N>
ostream& operator<<(ostream& os, const array<T, N> &arr) {
  for(const T &X : arr)
    os << X << ' ';
  return os;
}
template<class T>
ostream& operator<<(ostream& os, const vector<T> &vec) {
  for(const T &X : vec)
    os << X << ' ';
  return os;
}
template<class T>
ostream& operator<<(ostream& os, const set<T> &s) {
  for(const T &x : s)
    os << x << ' ';
  return os;
}

struct DSU {
  vector<int> bos, sz;
  int size;

  DSU(int _size) : bos(_size), sz(_size, 1), size(_size) {
    iota(bos.begin(), bos.end(), 0);
  }

  int query(int v) {
    if (bos[v] == v)
      return v;
    else
      return bos[v] = query(bos[v]);
  }

  bool merge(int v1, int v2) {
    int b1 = query(v1), b2 = query(v2);

    if (b1 == b2)
      return false;

    if (sz[b1] > sz[b2])
      swap(b1, b2);
    bos[b1] = b2, sz[b2] += sz[b1];

    return true;
  }
};

signed main() {
  ios::sync_with_stdio(false), cin.tie(NULL);

  int n, k, c; cin >> n >> k >> c;
  vector<array<int, 4>> e(k);
  for(auto &[u, v, w, p] : e) {
    cin >> u >> v >> w >> p;
    u--, v--;
  }

  R::sort(e, less<int>{}, [](auto x) { return x[2]; });

  vector<bool> inT(k, false);
  vector<int> teid;
  int cost = 0, price = 0;
  {
    DSU dsu(n);
    for(int i = 0; auto [u, v, w, p] : e) {
      if (dsu.merge(u, v)) {
        inT[i] = true, cost += w, price = max(price, p);
        teid.emplace_back(i);
      }
      i++;
    }
  }

  if (cost > c) {
    cout << -1 << '\n';
    return 0;
  }

  vector<array<int, 3>> bs;
  for(int i = 0; i < k; i++)
    if (!inT[i])
      bs.push_back({0, n - 2, i});

  while(!bs.empty()) {
    DSU dsu(n);
    sort(bs.begin(), bs.end(), [](auto x, auto y) { return (x[0] + x[1]) / 2 < (y[0] + y[1]) / 2; });
    vector<array<int, 3>> nxt;
    int ptr = 0;
    for(auto [l, r, i] : bs) {
      if (l == r) {
        if (int tmp = cost - e[teid[l]][2] + e[i][2]; tmp <= c)
          price = max(price, e[i][3]);
        continue;
      }
      int mid = (l + r) / 2;
      while(ptr <= mid) {
        int k = teid[ptr++];
        dsu.merge(e[k][0], e[k][1]);
      }
      if (dsu.query(e[i][0]) == dsu.query(e[i][1]))
        nxt.push_back({l, mid, i});
      else
        nxt.push_back({mid + 1, r, i});
    }
    bs.swap(nxt);
  }

  cout << price << '\n';

  return 0;
}
0