結果

問題 No.416 旅行会社
ユーザー OUDONOUDON
提出日時 2016-12-13 03:00:05
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 607 ms / 4,000 ms
コード長 2,477 bytes
コンパイル時間 4,133 ms
コンパイル使用メモリ 186,620 KB
実行使用メモリ 38,008 KB
最終ジャッジ日時 2023-08-21 10:09:24
合計ジャッジ時間 9,706 ms
ジャッジサーバーID
(参考情報)
judge13 / judge11
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 254 ms
23,844 KB
testcase_01 AC 2 ms
4,376 KB
testcase_02 AC 2 ms
4,380 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,380 KB
testcase_07 AC 3 ms
4,380 KB
testcase_08 AC 7 ms
4,380 KB
testcase_09 AC 34 ms
5,376 KB
testcase_10 AC 273 ms
23,812 KB
testcase_11 AC 303 ms
28,588 KB
testcase_12 AC 310 ms
28,592 KB
testcase_13 AC 262 ms
28,424 KB
testcase_14 AC 584 ms
34,720 KB
testcase_15 AC 579 ms
37,392 KB
testcase_16 AC 586 ms
38,008 KB
testcase_17 AC 607 ms
36,692 KB
testcase_18 AC 580 ms
36,640 KB
testcase_19 AC 435 ms
28,392 KB
testcase_20 AC 433 ms
25,560 KB
権限があれば一括ダウンロードができます

ソースコード

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;

        set<int> g;
        if (UF.is_same(0, s)) g = UF.get_same_group(t);
        if (UF.is_same(0, t)) g = UF.get_same_group(s);
        UF.merge(s, t);

        for (auto u : g) {
            assert(ans[u] == 0);
            ans[u] = Q - i;
        }
    }

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