結果

問題 No.416 旅行会社
ユーザー iqueue02iqueue02
提出日時 2019-12-25 17:55:27
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 623 ms / 4,000 ms
コード長 1,819 bytes
コンパイル時間 2,303 ms
コンパイル使用メモリ 182,876 KB
実行使用メモリ 19,920 KB
最終ジャッジ日時 2023-08-21 10:28:08
合計ジャッジ時間 9,208 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 280 ms
14,416 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 1 ms
4,376 KB
testcase_03 AC 2 ms
4,376 KB
testcase_04 AC 2 ms
4,380 KB
testcase_05 AC 2 ms
4,376 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 3 ms
4,380 KB
testcase_08 AC 8 ms
4,380 KB
testcase_09 AC 33 ms
4,592 KB
testcase_10 AC 303 ms
14,240 KB
testcase_11 AC 305 ms
14,348 KB
testcase_12 AC 301 ms
14,424 KB
testcase_13 AC 276 ms
14,356 KB
testcase_14 AC 602 ms
19,916 KB
testcase_15 AC 602 ms
19,844 KB
testcase_16 AC 615 ms
19,840 KB
testcase_17 AC 610 ms
19,920 KB
testcase_18 AC 623 ms
19,844 KB
testcase_19 AC 439 ms
17,888 KB
testcase_20 AC 444 ms
17,888 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i,n) for(int i = 0; i < (n);i++)
#define sz(x) int(x.size())
typedef long long ll;
typedef pair<int,int> P;

struct QuickFind {
  vector<int> i2g;
  vector<vector<int>> g2i;

  QuickFind (int n) {
   i2g.assign(n,-1);
   g2i.assign(n, vector<int>());
   for (int i = 0; i < n; i++) {
     i2g[i] = i;
     g2i[i].emplace_back(i);
   }
  }

  bool same(int ia, int ib) {return i2g[ia] == i2g[ib];}

  void merge(int ia, int ib) {
    if (same(ia, ib)) return ;

    if (g2i[i2g[ia]].size() < g2i[i2g[ib]].size()) {
      swap(ia, ib);
    }
    int ga = i2g[ia], gb = i2g[ib];
    for (auto u : g2i[gb]) {
      i2g[u] = ga;
    }
    g2i[ga].insert(g2i[ga].end(), g2i[gb].begin(), g2i[gb].end());
    g2i[gb].clear();
  }
};

int main(){
  int n, m, q;
  cin >> n >> m >> q;
  
  QuickFind qf(n);
  set<P> st;
  rep(i,m) {
    int a, b;
    cin >> a >> b;
    a--; b--;
    st.insert(make_pair(a, b));
  }
  
  vector<P> query(q);
  rep(i,q) {
    int c, d;
    cin >> c >> d;
    c--; d--;
    query[i] = make_pair(c, d);
    st.erase(make_pair(c, d));
  }
  reverse(query.begin(), query.end());
  for (auto e : st) qf.merge(e.second, e.first);

  vector<int> res(n, 0);
  for (auto e : qf.g2i[qf.i2g[0]]) res[e] = -1;

  for (int i = 0; i < q; i++) {
    int u = query[i].first, v = query[i].second;
    if (!qf.same(u, v)) {
      if (!qf.same(0,u) && !qf.same(0,v)) {
        qf.merge(u, v);
        continue;
      }
      if (qf.same(0, u)) {
        for (auto e : qf.g2i[qf.i2g[v]]) {
          if (res[e] == 0) res[e] = q - i;
        }
      } else {
        for (auto e : qf.g2i[qf.i2g[u]]) {
          if (res[e] == 0) res[e] = q - i;
        }
      }
    }
    qf.merge(u, v);
  }
  for (int i = 1; i < n; i++) cout << res[i] << endl;
  return 0;
}
0