結果

問題 No.2630 Colorful Vertices and Cheapest Paths
コンテスト
ユーザー Misuki
提出日時 2024-02-16 21:49:49
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 250 ms / 2,500 ms
コード長 3,485 bytes
コンパイル時間 2,385 ms
コンパイル使用メモリ 206,192 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-09-28 20:04:29
合計ジャッジ時間 6,413 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 22
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#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, m; cin >> n >> m;
  vector<array<int, 2>> e(m);
  for(auto &[u, v] : e) {
    cin >> u >> v;
    u--, v--;
  }
  vector<int> c(n);
  for(int &x : c) {
    cin >> x;
    x--;
  }
  array<int, 10> w;
  for(int &x : w)
    cin >> x;
  int q; cin >> q;
  vector<array<int, 2>> query(q);
  for(auto &[u, v] : query) {
    cin >> u >> v;
    u--, v--;
  }

  vector<ll> ans(q, LLONG_MAX);
  for(int msk = 0; msk < (1 << 10); msk++) {
    ll cost = 0;
    for(int i = 0; i < 10; i++)
      if (msk >> i & 1)
        cost += w[i];
    DSU dsu(n);
    for(auto [u, v] : e)
      if ((msk >> c[u] & 1) and (msk >> c[v] & 1))
        dsu.merge(u, v);
    for(int i = 0; auto [u, v] : query) {
      if ((msk >> c[u] & 1) and (msk >> c[v] & 1) and dsu.query(u) == dsu.query(v))
        ans[i] = min(ans[i], cost);
      i++;
    }
  }

  for(ll x : ans)
    cout << (x == LLONG_MAX ? -1 : x) << '\n';

  return 0;
}
0