結果

問題 No.922 東北きりきざむたん
ユーザー risujiroh
提出日時 2019-11-09 00:23:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 145 ms / 2,000 ms
コード長 3,241 bytes
コンパイル時間 2,190 ms
コンパイル使用メモリ 188,892 KB
実行使用メモリ 49,732 KB
最終ジャッジ日時 2024-09-15 03:15:16
合計ジャッジ時間 5,390 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

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