結果

問題 No.416 旅行会社
ユーザー mamekin
提出日時 2016-10-05 17:21:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 897 ms / 4,000 ms
コード長 3,342 bytes
コンパイル時間 1,780 ms
コンパイル使用メモリ 129,272 KB
実行使用メモリ 24,572 KB
最終ジャッジ日時 2024-12-14 20:20:24
合計ジャッジ時間 10,943 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define _USE_MATH_DEFINES
#include <cstdio>
#include <iostream>
#include <sstream>
#include <fstream>
#include <iomanip>
#include <algorithm>
#include <cmath>
#include <complex>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <set>
#include <map>
#include <bitset>
#include <numeric>
#include <limits>
#include <climits>
#include <cfloat>
#include <functional>
#include <iterator>
using namespace std;

class UnionFindTree
{
private:
    const int n;
    vector<int> parent;       // 親ノード
    vector<set<int> > child;  // 子ノード
    vector<int> rank;         // 木の高さの上限
    int find(int i){
        if(parent[i] == i){
            return i;
        }
        else{
            child[parent[i]].erase(i);
            parent[i] = find(parent[i]);
            child[parent[i]].insert(i);
            return parent[i];
        }
    }
public:
    UnionFindTree(int n) : n(n){ // コンストラクタ
        parent.resize(n);
        child.resize(n);
        for(int i=0; i<n; ++i){
            parent[i] = i;
            child[i].insert(i);
        }
        rank.assign(n, 0);
    }
    void unite(int a, int b){ // aとbのグループを併合
        if((a = find(a)) != (b = find(b))){
            if(rank[a] < rank[b]){
                parent[a] = b;
                child[b].insert(a);
            }
            else{
                parent[b] = a;
                child[a].insert(b);
                if(rank[a] == rank[b])
                    ++ rank[a];
            }
        }
    }
    bool same(int a, int b){ // aとbのグループが同じかを調べる
        return find(a) == find(b);
    }
    void getGroup(int a, vector<int>& group){ // グループ構成を返す
        set<int> s;
        queue<int> q;
        s.insert(a);
        q.push(a);
        while(!q.empty()){
            int x = q.front();
            q.pop();
            if(s.find(parent[x]) == s.end()){
                s.insert(parent[x]);
                q.push(parent[x]);
            }
            for(int y : child[x]){
                if(s.find(y) == s.end()){
                    s.insert(y);
                    q.push(y);
                }
            }
        }
        group = vector<int>(s.begin(), s.end());
    }
};

int main()
{
    int n, m, q;
    cin >> n >> m >> q;

    set<pair<int, int> > s;
    for(int i=0; i<m; ++i){
        int a, b;
        cin >> a >> b;
        -- a;
        -- b;
        s.insert(make_pair(a, b));
    }

    vector<int> c(q), d(q);
    for(int i=0; i<q; ++i){
        cin >> c[i] >> d[i];
        -- c[i];
        -- d[i];
        s.erase(make_pair(c[i], d[i]));
    }

    UnionFindTree uft(n);
    for(const auto& p : s)
        uft.unite(p.first, p.second);

    vector<int> ans(n, 0);
    vector<int> v;
    uft.getGroup(0, v);
    for(unsigned i=0; i<v.size(); ++i)
        ans[v[i]] = -1;

    for(int i=q-1; i>=0; --i){
        if(uft.same(c[i], d[i]))
            continue;

        vector<int> v;
        if(uft.same(0, c[i]))
            uft.getGroup(d[i], v);
        else if(uft.same(0, d[i]))
            uft.getGroup(c[i], v);
        for(unsigned j=0; j<v.size(); ++j)
            ans[v[j]] = i + 1;

        uft.unite(c[i], d[i]);
    }
    for(int i=1; i<n; ++i)
        cout << ans[i] << endl;

    return 0;
}
0