結果

問題 No.1194 Replace
ユーザー MayimgMayimg
提出日時 2020-08-25 02:50:46
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 980 ms / 2,000 ms
コード長 3,051 bytes
コンパイル時間 2,943 ms
コンパイル使用メモリ 220,632 KB
実行使用メモリ 103,364 KB
最終ジャッジ日時 2024-04-24 04:06:53
合計ジャッジ時間 16,568 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 516 ms
68,228 KB
testcase_01 AC 573 ms
71,864 KB
testcase_02 AC 448 ms
59,764 KB
testcase_03 AC 327 ms
51,444 KB
testcase_04 AC 562 ms
71,872 KB
testcase_05 AC 492 ms
67,300 KB
testcase_06 AC 453 ms
63,544 KB
testcase_07 AC 980 ms
103,240 KB
testcase_08 AC 867 ms
103,364 KB
testcase_09 AC 823 ms
103,248 KB
testcase_10 AC 827 ms
103,248 KB
testcase_11 AC 821 ms
103,256 KB
testcase_12 AC 827 ms
103,256 KB
testcase_13 AC 436 ms
42,052 KB
testcase_14 AC 337 ms
34,204 KB
testcase_15 AC 328 ms
40,112 KB
testcase_16 AC 423 ms
42,528 KB
testcase_17 AC 289 ms
36,676 KB
testcase_18 AC 274 ms
31,216 KB
testcase_19 AC 408 ms
44,752 KB
testcase_20 AC 2 ms
5,376 KB
testcase_21 AC 2 ms
5,376 KB
testcase_22 AC 2 ms
5,376 KB
testcase_23 AC 108 ms
20,032 KB
testcase_24 AC 30 ms
10,240 KB
testcase_25 AC 13 ms
5,376 KB
testcase_26 AC 120 ms
15,604 KB
testcase_27 AC 18 ms
5,376 KB
testcase_28 AC 52 ms
9,336 KB
testcase_29 AC 6 ms
5,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

map<int, int> zip;
vector<int> unzip;
template <typename T> void compress(vector<T>& v) {
  for (int i = 0; i < (int) v.size(); i++) unzip.push_back(v[i]);
  sort(unzip.begin(), unzip.end());
  unzip.erase(unique(unzip.begin(), unzip.end()), unzip.end());
  for (int i = 0; i < (int) unzip.size(); i++) zip[unzip[i]] = i;
}

class StronglyConnectedComponents {
private:
  int siz;
  int itr = 0;
  vector<bool> done;
  vector<int> order, comp;
  void dfs (int cur, const vector<vector<int>>& g) {
    done[cur] = true;
    for (int nxt : g[cur]) {
      if (!done[nxt]) dfs(nxt, g);
    }
    order.push_back(cur);
  }
  void rdfs (int cur, const vector<vector<int>>& rg) {
    done[cur] = true;
    comp[cur] = itr;
    for (int nxt : rg[cur]) {
      if (!done[nxt]) rdfs(nxt, rg);
    }
  }
  void build (const vector<vector<int>>& g, const vector<vector<int>>& rg) {
    for (int i = 0; i < siz; ++i) if (!done[i]) dfs (i, g);
    reverse(order.begin(), order.end());
    fill(done.begin(), done.end(), false);
    for (int i = 0; i < siz; ++i) if (!done[order[i]]) rdfs (order[i], rg), ++itr;
  }

public:
  StronglyConnectedComponents(const vector<vector<int>>& g, const vector<vector<int>>& rg, const int& n) : siz(n), done(siz, false), comp(siz, -1) {
    build(g, rg);
  }
  bool  has_cycle() { return *max_element(comp.begin(), comp.end()) < siz - 1; }
  vector<int> get_comp() { return comp;}
};
using SCC = StronglyConnectedComponents;

int dfs (int cur, vector<set<int>>& g, vector<int>& val, vector<int>& vis) {
  if (vis[cur]) return val[cur];
  vis[cur] = 1;
  for (int nxt : g[cur]) {
    val[cur] = max(val[cur], dfs(nxt, g, val, vis));
  }
  return val[cur];
}

signed main() { 
  ios::sync_with_stdio(false); cin.tie(0);
  int n, m;
  cin >> n >> m;
  vector<int> all;
  vector<int> b(m), c(m);
  for (int i = 0; i < m; i++) {
    cin >> b[i] >> c[i];
    all.push_back(b[i]);
    all.push_back(c[i]);
  }
  compress(all);
  for (int i = 0; i < m; i++) {
    b[i] = zip[b[i]];
    c[i] = zip[c[i]];
  }
  int nn = zip.size();
  vector<vector<int>> g(nn), rg(nn);
  for (int i = 0; i < m; i++) {
    g[b[i]].push_back(c[i]);
    rg[c[i]].push_back(b[i]);
  }
  SCC scc(g, rg, nn);
  vector<int> comp = scc.get_comp();
  int nnn = *max_element(comp.begin(), comp.end()) + 1;
  vector<set<int>> gg(nnn);
  vector<int> val(nnn);
  for (int i = 0; i < m; i++) {
    assert(comp[b[i]] < nnn);
    assert(comp[c[i]] < nnn);
    val[comp[b[i]]] = max(val[comp[b[i]]], unzip[b[i]]);
    val[comp[c[i]]] = max(val[comp[c[i]]], unzip[c[i]]);
    if (comp[b[i]] == comp[c[i]]) continue;
    gg[comp[b[i]]].insert(comp[c[i]]);
  }
  vector<int> vis(nnn);
  for (int i = 0; i < nnn; i++) dfs (i, gg, val, vis);
  long long ans = 1LL * n * (n + 1) / 2;
  set<int> app;
  for (auto x : b) {
    if (app.count(x)) continue;
    app.insert(x);
    int y = unzip[x];
    ans -= y;
    assert(y <= val[comp[x]]);
    ans += val[comp[x]];
  }
  cout << ans << endl;
  return 0;
}
0