結果

問題 No.416 旅行会社
ユーザー momoyuumomoyuu
提出日時 2022-10-24 14:51:06
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 538 ms / 4,000 ms
コード長 2,879 bytes
コンパイル時間 3,554 ms
コンパイル使用メモリ 258,516 KB
実行使用メモリ 15,232 KB
最終ジャッジ日時 2024-05-08 16:09:30
合計ジャッジ時間 9,561 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 152 ms
9,472 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 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 5 ms
5,376 KB
testcase_09 AC 20 ms
5,376 KB
testcase_10 AC 177 ms
9,472 KB
testcase_11 AC 175 ms
9,472 KB
testcase_12 AC 176 ms
9,472 KB
testcase_13 AC 154 ms
11,008 KB
testcase_14 AC 457 ms
14,976 KB
testcase_15 AC 472 ms
15,104 KB
testcase_16 AC 538 ms
15,232 KB
testcase_17 AC 521 ms
15,104 KB
testcase_18 AC 460 ms
15,104 KB
testcase_19 AC 290 ms
12,544 KB
testcase_20 AC 298 ms
12,544 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using ll = long long;
using ull = unsigned long long;
using ld = long double;
template<class t> using vc = vector<t>;
template<class t> using vvc = vc<vc<t>>;
using pi = pair<int,int>;
using pl = pair<ll,ll>;
using vi = vc<int>;
using vvi = vvc<int>;
using vl = vc<ll>;
using vvl = vvc<ll>;

#define rep(i,a,b) for (int i = (int)(a); i < (int)(b); i++)
#define irep(i,a,b) for (int i = (int)(a); i > (int)(b); i--)
#define all(a) a.begin(),a.end()
#define print(n) cout << n << '\n'
#define pritn(n) print(n)
#define printv(n,a) {copy(all(n),ostream_iterator<a>(cout," ")); cout<<"\n";}
#define printvv(n,a) {for(auto itr:n) printv(itr,a);}
#define rup(a,b) (a+b-1)/b
#define input(A,N) rep(i,0,N) cin>>A[i]
#define chmax(a,b) a = max(a,b)
#define chmin(a,b) a = min(a,b)

struct unionfind{
    int n;
    vector<int> parent;
    vector<int> num;
    
    unionfind(int n):n(n),parent(vector<int>(n)),num(vector<int>(n,1)){
        for(int i = 0; i < n; i++) parent[i] = i;
    }

    int find(int n){
        if (parent[n] == n) return n;
        parent[n] = find(parent[n]);
        return parent[n];
    }

    int merge(int n,int m){
        n = find(n);
        m = find(m);
        if (n==m) return n;
        if (num[n]>num[m]) return merge(m,n);
        parent[m] = n;
        num[n] += num[m];
        return n;
    }
    
    int getnum(int n){
        return num[find(n)];
    }
};

int dfs(vi&ans,vi&now,int ni){
    if(ans[now[ni]] != -2){
        ans[ni] = ans[now[ni]];
        return ans[ni];
    }
    return ans[ni] = dfs(ans,now,now[ni]);
}

int main(){
    cout << fixed << setprecision(15);
    
    int n,m,q;
    cin>>n>>m>>q;
    unionfind uf(n);
    set<pi> s;
    rep(i,0,m){
        int a,b;
        cin>>a>>b;
        a--;b--;
        s.insert(pi(a,b));
    }
    vi c(q),d(q);
    rep(i,0,q){
        cin>>c[i]>>d[i];
        c[i]--;d[i]--;
    }
    
    rep(i,0,q)s.erase(pi(c[i],d[i]));

    for(auto now:s){
        uf.merge(now.first,now.second);
    }

    vi ans(n,-2);
    vi now(n,0);
    rep(i,0,n) now[i] = uf.find(i);

    rep(i,0,n){
        if(uf.find(i)==uf.find(0)) ans[i] = -1;
    }

    irep(i,q-1,-1){
        int ni = c[i];
        int mi = d[i];
        ni = uf.find(ni);
        mi = uf.find(mi);
        if(ni==mi) continue;
        if(ni!=uf.find(0)&&mi!=uf.find(0)){
            uf.merge(ni,mi);
            now[mi] = uf.find(mi);
            now[ni] = uf.find(ni);
            continue;
        }
        if(mi==uf.find(0)) swap(ni,mi);
        ans[mi] = i + 1;
        now[mi] = ni;
        uf.merge(ni,mi);
    }
    ans[0] = 0;
    rep(i,0,n){
        if(ans[i]!=-2) continue;
        if(uf.find(i)!=uf.find(0)){
            ans[i] = 0;
            continue;
        }
        dfs(ans,now,i);
    }
    rep(i,1,n) print(ans[i]);

    
    //system("pause");
    return 0;
}
0