#599562 (C++17) No.317 辺の追加

提出ソース

結果

問題	No.317 辺の追加
ユーザー	0w1
提出日時	2021-01-05 03:35:44
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	600 ms / 2,000 ms
コード長	4,943 bytes
コンパイル時間	2,511 ms
コンパイル使用メモリ	221,876 KB
実行使用メモリ	5,248 KB
最終ジャッジ日時	2024-10-15 12:16:44
合計ジャッジ時間	11,989 ms
ジャッジサーバーID （参考情報）	judge3 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	2 ms 5,248 KB
testcase_02	AC	45 ms 5,248 KB
testcase_03	AC	46 ms 5,248 KB
testcase_04	AC	78 ms 5,248 KB
testcase_05	AC	44 ms 5,248 KB
testcase_06	AC	54 ms 5,248 KB
testcase_07	AC	18 ms 5,248 KB
testcase_08	AC	202 ms 5,248 KB
testcase_09	AC	59 ms 5,248 KB
testcase_10	AC	67 ms 5,248 KB
testcase_11	AC	31 ms 5,248 KB
testcase_12	AC	65 ms 5,248 KB
testcase_13	AC	52 ms 5,248 KB
testcase_14	AC	60 ms 5,248 KB
testcase_15	AC	65 ms 5,248 KB
testcase_16	AC	72 ms 5,248 KB
testcase_17	AC	60 ms 5,248 KB
testcase_18	AC	67 ms 5,248 KB
testcase_19	AC	66 ms 5,248 KB
testcase_20	AC	98 ms 5,248 KB
testcase_21	AC	26 ms 5,248 KB
testcase_22	AC	15 ms 5,248 KB
testcase_23	AC	17 ms 5,248 KB
testcase_24	AC	39 ms 5,248 KB
testcase_25	AC	31 ms 5,248 KB
testcase_26	AC	34 ms 5,248 KB
testcase_27	AC	29 ms 5,248 KB
testcase_28	AC	18 ms 5,248 KB
testcase_29	AC	5 ms 5,248 KB
testcase_30	AC	504 ms 5,248 KB
testcase_31	AC	586 ms 5,248 KB
testcase_32	AC	588 ms 5,248 KB
testcase_33	AC	592 ms 5,248 KB
testcase_34	AC	586 ms 5,248 KB
testcase_35	AC	595 ms 5,248 KB
testcase_36	AC	592 ms 5,248 KB
testcase_37	AC	600 ms 5,248 KB
testcase_38	AC	502 ms 5,248 KB
testcase_39	AC	591 ms 5,248 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;

/*^ debug  */

template <typename A, typename B> string to_string(pair<A, B> p);
template <typename A, typename B, typename C> string to_string(tuple<A, B, C> p);
template <typename A, typename B, typename C, typename D> string to_string(tuple<A, B, C, D> p);
string to_string(const string& s) { return '"' + s + '"'; }
string to_string(const char* s) { return to_string((string) s); }
string to_string(bool b) { return (b ? "true" : "false"); }
string to_string(vector<bool> v) {
  bool first = true;
  string res = "{";
  for (int i = 0; i < static_cast<int>(v.size()); i++) {
    if (!first) { res += ", "; }
    first = false;
    res += to_string(v[i]);
  }
  res += "}";
  return res;
}
template <size_t N>
string to_string(bitset<N> v) {
  string res = "";
  for (size_t i = 0; i < N; i++) { res += static_cast<char>('0' + v[i]); }
  return res;
}
template <typename A>
string to_string(A v) {
  bool first = true;
  string res = "{";
  for (const auto &x : v) {
    if (!first) { res += ", "; }
    first = false;
    res += to_string(x);
  }
  res += "}";
  return res;
}
template <typename A, typename B>
string to_string(pair<A, B> p) { return "(" + to_string(p.first) + ", " + to_string(p.second) + ")"; }
template <typename A, typename B, typename C>
string to_string(tuple<A, B, C> p) { return "(" + to_string(get<0>(p)) + ", " + to_string(get<1>(p)) + ", " + to_string(get<2>(p)) + ")"; }
template <typename A, typename B, typename C, typename D>
string to_string(tuple<A, B, C, D> p) { return "(" + to_string(get<0>(p)) + ", " + to_string(get<1>(p)) + ", " + to_string(get<2>(p)) + ", " + to_string(get<3>(p)) + ")"; }
void debug_out() { cerr << endl; }
template <typename Head, typename... Tail>
void debug_out(Head H, Tail... T) { cerr << " " << to_string(H); debug_out(T...); }
#ifdef LOCAL
#define debug(...) cerr << "[" << #__VA_ARGS__ << "]:", debug_out(__VA_ARGS__)
#else
#define debug(...) 42
#endif

/*  debug $*/

/*^ vector extensions  */

template<typename T>
T concat(initializer_list<T> lists) {
  T a;
  for (auto &l : lists) a.insert(a.end(), l.begin(), l.end());
  return a;
}

template<typename T, size_t sz>
struct _Matrix_type { typedef vector<typename _Matrix_type<T, sz - 1>::type> type; };
template<typename T>
struct _Matrix_type<T, 1> { typedef T type; };
template<typename T>
struct _Matrix {
  static auto build(size_t s) { return vector<T>(s); }
  template<typename ...Args>
  static auto build(size_t f, Args... args) {
    return vector<typename _Matrix_type<T, 1 + sizeof...(args)>::type>(f, _Matrix<T>::build(args...));
  }
};
template<typename T, typename... Args>
auto buildMatrix(Args... args) { return _Matrix<T>::build(args...); }

/*  vector extensions $*/

/*^ generic definitions  */

template<typename F>
struct _RecurFun : F {
  _RecurFun(F&& f) : F(forward<F>(f)) {}
  template<typename... Args>
  decltype(auto) operator()(Args&&... args) const { return F::operator()(*this, forward<Args>(args)...); }
};
template<typename F>
decltype(auto) RecurFun(F&& f) { return _RecurFun<F> { forward<F>(f) }; }

/*  generic definitions $*/

struct DSU {
  vector<int> fa;
  vector<int> sz;

  DSU(int n) : fa(n), sz(n, 1) { iota(fa.begin(), fa.end(), 0); }

  int find(int x) { return fa[x] == x ? x : fa[x] = find(fa[x]); }

  bool unite(int x, int y) {
    int fx = find(x);
    int fy = find(y);
    if (fx == fy) return false;
    fa[fy] = fx;
    sz[fx] += sz[fy];
    return true;
  }
};

template<typename T, T INF = 0x3f3f3f3f>
vector<T> multiKnapsack(const int W, const vector<int> &ws, const vector<T> &vs, const vector<int> &cs) {
  const int n = ws.size();
  vector<T> dp(W + 1, -INF); {
    dp[0] = 0;
    for (int i = 0; i < n; ++i) {
      for (int r = 0; r < ws[i]; ++r) {
        deque<pair<T, int>> dq;
        for (int k = r; k <= W; k += ws[i]) {
          while (dq.size() && (k - dq.front().second) > ws[i] * cs[i]) dq.pop_front();
          while (dq.size() && dq.back().first <= dp[k] - vs[i] * (k / ws[i] + 1)) dq.pop_back();
          dq.emplace_back(dp[k] - vs[i] * (k / ws[i] + 1), k);
          if (dq.front().second < k) {
            dp[k] = max(dp[k], dq.front().first + vs[i] * (k / ws[i] + 1));
          }
        }
      }
    }
  }
  return dp;
}

int main() {
  ios::sync_with_stdio(false);

  int N, M; { cin >> N >> M; }

  DSU dsu(N); {
    for (int i = 0; i < M; ++i) {
      int U, V; { cin >> U >> V; --U, --V; }
      dsu.unite(U, V);
    }
  }

  map<int, int> ccsize2cnt; {
    for (int u = 0; u < N; ++u) if (dsu.find(u) == u) {
      ccsize2cnt[dsu.sz[u]] += 1;
    }
  }

  vector<int> dp; {
    vector<int> ws, vs, cs;
    for (auto [k, v] : ccsize2cnt) {
      ws.push_back(k);
      vs.push_back(-1);
      cs.push_back(v);
    }
    dp = multiKnapsack(N, ws, vs, cs);
    for (int &v : dp) v = -v;
  }

  for (int i = 1; i <= N; ++i) {
    cout << (dp[i] > N ? -1 : dp[i] - 1) << "\n";
  }
}

yukicoder

結果

テストケース

ソースコード