結果

問題 No.416 旅行会社
ユーザー しらっ亭
提出日時 2016-08-28 01:05:00
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 310 ms / 4,000 ms
コード長 3,088 bytes
コンパイル時間 2,024 ms
コンパイル使用メモリ 183,772 KB
実行使用メモリ 24,044 KB
最終ジャッジ日時 2024-12-14 20:04:30
合計ジャッジ時間 6,038 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define _p(...) (void)printf(__VA_ARGS__)
#define forr(x,arr) for(auto&& x:arr)
#define _overload3(_1,_2,_3,name,...) name
#define _rep2(i,n) _rep3(i,0,n)
#define _rep3(i,a,b) for(int i=int(a);i<int(b);++i)
#define rep(...) _overload3(__VA_ARGS__,_rep3,_rep2,)(__VA_ARGS__)
#define _rrep2(i,n) _rrep3(i,0,n)
#define _rrep3(i,a,b) for(int i=int(b)-1;i>=int(a);i--)
#define rrep(...) _overload3(__VA_ARGS__,_rrep3,_rrep2,)(__VA_ARGS__)
#define ALL(x) (x).begin(), (x).end()
#define BIT(n) (1LL<<(n))
#define SZ(x) ((int)(x).size())
#define fst first
#define snd second
typedef long long ll;
typedef vector<int> vi;typedef vector<vi> vvi;typedef pair<int,int> pii;typedef vector<pii> vpii;

struct UnionFind {
  const int V; // 頂点数
  vector<int> par, cnt; // 親の番号, 同じ親の頂点数
  vector<vector<int>> g2i;
  UnionFind(int V) : V(V), par(V), cnt(V), g2i(V) { init(); }

  // 初期化する
  void init() {
    for (int i = 0; i < V; i++) {
      par[i] = i;
      g2i[i].assign(1, i);
    }
    fill(cnt.begin(), cnt.end(), 1);
  }

  // 木の根を求める
  int find(int x) {
    return par[x] == x ? x : par[x] = find(par[x]);
  }

  // x と y の属する集合を併合
  // 併合前は違う集合だったら true を返す
  bool unite(int x, int y) {
    if ((x = find(x)) == (y = find(y))) return false;

    // x の所属するグループが y の所属するグループより小さくならないようにする
    if (cnt[x] < cnt[y]) swap(x, y);

    // グループ y に所属する全てのアイテムをグループ x に移す
    g2i[x].insert(g2i[x].end(), g2i[y].begin(), g2i[y].end());
    g2i[y].clear();

    cnt[y] = cnt[x] = cnt[x] + cnt[y];
    par[y] = x;

    return true;
  }

  // x と y が同じ集合に属するか否か
  bool same(int x, int y) {
    return find(x) == find(y);
  }

  // x と同じ集合にいる頂点数を返す
  int count(int x) {
    return cnt[find(x)];
  }
};

void Main() {
  int N, M, Q;
  scanf("%d%d%d", &N, &M, &Q);

  vpii bridges;
  vpii breaks;
  set<pii> break_set;

  rep(i, M) {
    int u, v;
    scanf("%d%d", &u, &v);
    u--; v--;
    bridges.emplace_back(u, v);
  }

  rep(i, Q) {
    int u, v;
    scanf("%d%d", &u, &v);
    u--; v--;
    breaks.emplace_back(u, v);
    break_set.emplace(u, v);
  }

  vi ans(N);
  UnionFind uf(N);

  // 壊れない橋を繋ぐ
  forr(uv, bridges) {
    if (!break_set.count(uv)) {
      uf.unite(uv.fst, uv.snd);
    }
  }

  // 最初から到達できるやつ
  rep(i, 1, N) {
    if (uf.same(0, i)) {
      ans[i] = -1;
    }
  }

  // 逆回し
  rrep(i, Q) {
    int c = breaks[i].fst;
    int d = breaks[i].snd;
    int gc = uf.find(c);
    int gd = uf.find(d);
    int g0 = uf.find(0);

    if (gc == gd) continue;
    if (gc == g0) {
      forr(v, uf.g2i[gd]) ans[v] = i+1;
    }
    else if (gd == g0) {
      forr(v, uf.g2i[gc]) ans[v] = i+1;
    }
    uf.unite(c, d);
  }

  rep(i, 1, N) _p("%d\n", ans[i]);
}
int main() { cin.tie(0); ios::sync_with_stdio(false); Main(); return 0; }
0