結果

問題 No.922 東北きりきざむたん
ユーザー risujirohrisujiroh
提出日時 2019-11-09 00:23:07
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 147 ms / 2,000 ms
コード長 3,241 bytes
コンパイル時間 2,266 ms
コンパイル使用メモリ 187,612 KB
実行使用メモリ 49,736 KB
最終ジャッジ日時 2023-10-13 05:56:18
合計ジャッジ時間 6,852 ms
ジャッジサーバーID
(参考情報)
judge11 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,352 KB
testcase_01 AC 1 ms
4,348 KB
testcase_02 AC 2 ms
4,352 KB
testcase_03 AC 1 ms
4,352 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 2 ms
4,348 KB
testcase_06 AC 2 ms
4,348 KB
testcase_07 AC 2 ms
4,352 KB
testcase_08 AC 2 ms
4,352 KB
testcase_09 AC 45 ms
20,116 KB
testcase_10 AC 25 ms
8,524 KB
testcase_11 AC 41 ms
16,704 KB
testcase_12 AC 21 ms
17,684 KB
testcase_13 AC 11 ms
6,828 KB
testcase_14 AC 70 ms
28,940 KB
testcase_15 AC 18 ms
18,116 KB
testcase_16 AC 114 ms
38,604 KB
testcase_17 AC 115 ms
38,668 KB
testcase_18 AC 117 ms
38,536 KB
testcase_19 AC 113 ms
38,664 KB
testcase_20 AC 110 ms
38,664 KB
testcase_21 AC 135 ms
40,812 KB
testcase_22 AC 147 ms
40,764 KB
testcase_23 AC 123 ms
40,208 KB
testcase_24 AC 128 ms
39,984 KB
testcase_25 AC 107 ms
40,920 KB
testcase_26 AC 111 ms
40,768 KB
testcase_27 AC 110 ms
40,888 KB
testcase_28 AC 38 ms
21,684 KB
testcase_29 AC 110 ms
49,736 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using lint = long long;
template<class T = int> using V = vector<T>;
template<class T = int> using VV = V< V<T> >;

struct UnionFind {
  V<> t;
  UnionFind(int n) : t(n, -1) {}
  int find(int v) { return t[v] < 0 ? v : t[v] = find(t[v]); }
  void unite(int u, int v) {
    u = find(u);
    v = find(v);
    if (u == v) return;
    if (-t[u] < -t[v]) swap(u, v);
    t[u] += t[v];
    t[v] = u;
  }
  int size(int v) { return -t[find(v)]; }
};

template<class T> struct SparseTable {
  VV<T> t;
  SparseTable(const V<T>& a) : t(1, a) {
    for (int k = 1, n = a.size(); 1 << k <= n; ++k) {
      t.emplace_back(n - (1 << k) + 1);
      for (int i = 0; i + (1 << k) <= n; ++i) {
        t[k][i] = min(t[k - 1][i], t[k - 1][i + (1 << (k - 1))]);
      }
    }
  }
  T acc(int l, int r) const {
    assert(l < r);
    int k = __lg(r - l);
    return min(t[k][l], t[k][r - (1 << k)]);
  }
};

int main() {
  cin.tie(nullptr); ios::sync_with_stdio(false);
  int n, m, q; cin >> n >> m >> q;
  UnionFind uf(n);
  VV<> g(n);
  while (m--) {
    int u, v; cin >> u >> v, --u, --v;
    uf.unite(u, v);
    g[u].push_back(v);
    g[v].push_back(u);
  }
  V<> dep(n), vs;
  {
    V<bool> vis(n);
    auto dfs = [&](auto& F, int v, int p) -> void {
      vis[v] = true;
      vs.push_back(v);
      for (int u : g[v]) if (u != p) {
        dep[u] = dep[v] + 1;
        F(F, u, v);
        vs.push_back(v);
      }
    };
    for (int v = 0; v < n; ++v) if (not vis[v]) {
      dfs(dfs, v, -1);
    }
  }
  V< pair<int, int> > _st;
  V<> id(n, 1e9);
  for (int i = 0; i < (int)vs.size(); ++i) {
    _st.emplace_back(dep[vs[i]], vs[i]);
    id[vs[i]] = min(id[vs[i]], i);
  }
  SparseTable< pair<int, int> > st(_st);
  V<> cum(vs.size() + 1);
  for (int i = (int)vs.size() - 2; i >= 0; --i) {
    int d = dep[vs[i]] < dep[vs[i + 1]] ? 1 : -1;
    cum[i] = d + cum[i + 1];
  }
  auto lca = [&](int u, int v) -> int {
    if (u == v) return u;
    tie(u, v) = minmax(id[u], id[v]);
    return st.acc(u, v).second;
  };
  auto len = [&](int u, int v) -> int {
    int a = lca(u, v);
    return 2 * cum[id[a]] - cum[id[u]] - cum[id[v]];
  };
  lint res = 0;
  V<> cnt(n);
  while (q--) {
    int u, v; cin >> u >> v, --u, --v;
    if (uf.find(u) == uf.find(v)) {
      res += len(u, v);
    } else {
      ++cnt[u];
      ++cnt[v];
    }
  }
  V<> sz(n);
  {
    V<bool> vis(n);
    auto dfs = [&](auto& F, int v, int p) -> void {
      vis[v] = true;
      sz[v] = cnt[v];
      for (int u : g[v]) if (u != p) {
        F(F, u, v);
        sz[v] += sz[u];
      }
    };
    for (int v = 0; v < n; ++v) if (not vis[v]) {
      dfs(dfs, v, -1);
    }
  }
  V<> cent(n, -1);
  {
    V<bool> vis(n);
    int csz = -1;
    auto dfs = [&](auto& F, int v, int p) -> void {
      vis[v] = true;
      bool ok = 2 * (csz - sz[v]) <= csz;
      for (int u : g[v]) if (u != p) {
        F(F, u, v);
        if (2 * sz[u] > csz) ok = false;
      }
      if (ok) {
        cent[uf.find(v)] = v;
      }
    };
    for (int v = 0; v < n; ++v) if (not vis[v]) {
      csz = sz[v];
      dfs(dfs, v, -1);
    }
  }
  for (int v = 0; v < n; ++v) {
    res += (lint)cnt[v] * len(v, cent[uf.find(v)]);
  }
  cout << res << '\n';
}
0