結果

問題 No.416 旅行会社
ユーザー iqueue02
提出日時 2019-12-25 17:55:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 582 ms / 4,000 ms
コード長 1,819 bytes
コンパイル時間 2,218 ms
コンパイル使用メモリ 185,592 KB
実行使用メモリ 19,968 KB
最終ジャッジ日時 2024-12-14 20:40:59
合計ジャッジ時間 9,269 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

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