結果

問題 No.416 旅行会社
ユーザー wk1080idwk1080id
提出日時 2019-03-17 22:03:52
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 626 ms / 4,000 ms
コード長 3,840 bytes
コンパイル時間 1,176 ms
コンパイル使用メモリ 102,976 KB
実行使用メモリ 32,500 KB
最終ジャッジ日時 2024-05-08 15:42:43
合計ジャッジ時間 8,449 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 307 ms
20,716 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 1 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 8 ms
5,376 KB
testcase_09 AC 34 ms
5,376 KB
testcase_10 AC 341 ms
20,720 KB
testcase_11 AC 333 ms
21,864 KB
testcase_12 AC 337 ms
21,864 KB
testcase_13 AC 305 ms
21,864 KB
testcase_14 AC 613 ms
31,184 KB
testcase_15 AC 613 ms
32,288 KB
testcase_16 AC 624 ms
32,500 KB
testcase_17 AC 613 ms
31,832 KB
testcase_18 AC 626 ms
31,488 KB
testcase_19 AC 415 ms
22,232 KB
testcase_20 AC 412 ms
21,504 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cctype>
#include<math.h>
#include<string>
#include<string.h>
#include<stack>
#include<queue>
#include<vector>
#include<utility>
#include<set>
#include<map>
#include<stdlib.h>
#include<iomanip>

using namespace std;

#define ll long long
#define ld long double
#define EPS 0.0000000001
#define INF 1e9
#define LINF (ll)INF*INF
#define MOD 1000000007
#define rep(i,n) for(int i=0;i<(n);i++)
#define loop(i,a,n) for(int i=a;i<(n);i++)
#define all(in) in.begin(),in.end()
#define shosu(x) fixed<<setprecision(x)

#define int ll //!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!

typedef vector<int> vi;
typedef vector<string> vs;
typedef pair<int,int> pii;
typedef vector<pii> vp;

int gcd(int a, int b){
    if(b==0) return a;
    return gcd(b,a%b);
}
int lcm(int a, int b){
    return a/gcd(a,b)*b;
}


//Quick Find Weighted
//https://topcoder.g.hatena.ne.jp/iwiwi/20131226/1388062106
//O(NlogN)

class UnionFind{

private:
    vector<int> i2g;          // i2g[i] := アイテム i の所属するグループの番号
    vector<vector<int> > g2i;  // g2i[g] := グループ g に所属するアイテムたち

public:
    UnionFind(){};
    UnionFind(int size){
        i2g.resize(size);
        g2i.resize(size);
        for (int i = 0; i < size; i++) {
        // 最初はアイテム i はグループ i に所属
            i2g[i] = i;
            g2i[i].assign(1, i);
        }
    }

    // アイテム ia の所属するグループとアイテム ib の所属するグループを 1 つにする
    // （ia と ib は違うグループに所属するものとする）
    void merge(int ia, int ib) {
        // ia の所属するグループが ib の所属するグループより小さくならないようにする
        if (g2i[i2g[ia]].size() < g2i[i2g[ib]].size()) {
            swap(ia, ib);
        }
        int ga = i2g[ia], gb = i2g[ib];
        // グループ gb に所属する全てのアイテムをグループ ga に移す
        for (int i = 0; i < g2i[gb].size(); i++) i2g[g2i[gb][i]] = ga;
        g2i[ga].insert(g2i[ga].end(), g2i[gb].begin(), g2i[gb].end());
        g2i[gb].clear();
    }

    // アイテム id と同じグループに所属しているアイテムを取得
    vector<int> get_group(int id){
        vector<int> ret;
        id = i2g[id];
        for(int i = 0; i < g2i[id].size(); i++)ret.push_back(g2i[id][i]);
        return ret;
    }

    // アイテム ia とアイテム ib は同じグループに所属しているか？
    bool same(int ia, int ib) {
        return i2g[ia] == i2g[ib];
    }
};

signed main(void) {
    int n,m,q;
    cin >> n >> m >> q;
    vp in(m),e(q);
    rep(i,m){
        int a,b;
        cin >> a >> b;
        a--,b--;
        in[i] = pii(a,b);
    }
    map<pii,int> ma;
    rep(i,q){
        int a,b;
        cin >> a >> b;
        a--,b--;
        e[i] = pii(a,b);
        ma[pii(a,b)]++;
    }

    vi ans(n,0);
    UnionFind uf(n);
    rep(i,m)if(ma.find(in[i]) == ma.end()){
        int a = in[i].first;
        int b = in[i].second;
        if(uf.same(a,b))continue;
        if(uf.same(a,0)){
            vi v = uf.get_group(b);
            rep(i,v.size())ans[v[i]] = -1;
        }else if(uf.same(b,0)){
            vi v = uf.get_group(a);
            rep(i,v.size())ans[v[i]] = -1;
        }
        uf.merge(a,b);
    }

    for(int i = q-1; i >= 0; i--){
        int a = e[i].first;
        int b = e[i].second;
        if(uf.same(a,b))continue;
        if(uf.same(a,0)){
            vi v = uf.get_group(b);
            rep(j,v.size())ans[v[j]] = i+1;
        }else if(uf.same(b,0)){
            vi v = uf.get_group(a);
            rep(j,v.size())ans[v[j]] = i+1;
        }
        uf.merge(a,b);
    }
    loop(i,1,n)cout << ans[i] << endl;
}
0