結果

問題 No.416 旅行会社
コンテスト
ユーザー OUDON
提出日時 2016-12-13 02:40:09
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,632 bytes
コンパイル時間 2,141 ms
コンパイル使用メモリ 190,764 KB
実行使用メモリ 53,156 KB
最終ジャッジ日時 2024-11-30 00:27:31
合計ジャッジ時間 45,968 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 13 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define DUMP(x) cerr << #x << "=" << x << endl
#define DUMP2(x, y) cerr<<"("<<#x<<", "<<#y<<") = ("<<x<<", "<<y<<")"<< endl
#define BINARY(x) static_cast<bitset<16> >(x)

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define REP(i,m,n) for (int i=m;i<(int)(n);i++)

#define in_range(x, y, w, h) (0<=(int)(x) && (int)(x)<(int)(w) && 0<=(int)(y) && (int)(y)<(int)(h))
#define ALL(a) (a).begin(),(a).end()

typedef long long ll;
const int INF = 1e9;
typedef pair<int, int> PII;
int dx[4]={0, -1, 1, 0}, dy[4]={-1, 0, 0, 1};


class UnionFind {
public:
    vector<int> par, union_size;
    vector<set<int>> group; // group[i] := 根iの集合に属する要素の一覧
    
    UnionFind(int n) : par(n), union_size(n, 1), group(n)
    {
        for (int i=0; i<n; i++) {
            par[i] = i;
            group[i].insert(i);
        }
    }
    
    // xとyの属する集合を併合する
    void merge(int x, int y)
    {
        x = find(x), y = find(y);
        if (x == y) return;
        if (union_size[x] < union_size[y]) swap(x, y);

        union_size[x] += union_size[y];
        union_size[y] = union_size[x];

        par[y] = x;
        group[x].insert(group[y].begin(), group[y].end());
        group[y] = set<int>();
    }
    
    set<int>& get_same_group(int x) { return group[find(x)]; }
    int find(int x) { return (par[x] == x ? x : par[x] = find(par[x])); }
    bool is_same(int x, int y) { return find(x) == find(y); }
};


int main()
{
    ios::sync_with_stdio(false);
    int N, M, Q;
    cin >> N >> M >> Q;
    UnionFind UF(N);

    set<PII> AS, BS; 
    vector<PII> B;
    rep(i, M + Q) {
        int U, V;
        cin >> U >> V;
        U--, V--;
        (i < M ? AS : BS).emplace(U, V);
        if (i >= M) B.emplace_back(U, V);
    }

    for (auto e : AS) {
        if (!BS.count(e)) UF.merge(e.first, e.second);
    }

    vector<int> ans(N);
    REP(i, 1, N) {
        ans[i] = (UF.is_same(0, i) ? -1 : 0);
    }
    UF.get_same_group(0) = set<int>();

    reverse(B.begin(), B.end());

    rep(i, Q) {
        int s = B[i].first, t = B[i].second;
        if (UF.is_same(s, t)) continue;
        UF.merge(s, t);

        set<int> g = UF.get_same_group(B[i].first);
        if (g.size() == 0) g = UF.get_same_group(B[i].second);

        if (UF.is_same(0, s) || UF.is_same(0, t)) {
            for (auto u : g) {
                assert(ans[u] == 0);
                ans[u] = Q - i;
            }
            UF.get_same_group(s) = set<int>();
            UF.get_same_group(t) = set<int>();
        }
    }

    REP(i, 1, N) cout << ans[i] << endl;
}
0