結果

問題 No.416 旅行会社
ユーザー outlineoutline
提出日時 2021-01-30 18:03:34
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 434 ms / 4,000 ms
コード長 3,374 bytes
コンパイル時間 1,804 ms
コンパイル使用メモリ 153,572 KB
実行使用メモリ 23,276 KB
最終ジャッジ日時 2024-05-08 16:02:18
合計ジャッジ時間 6,645 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 124 ms
14,572 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 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 19 ms
5,376 KB
testcase_10 AC 146 ms
14,572 KB
testcase_11 AC 133 ms
14,332 KB
testcase_12 AC 138 ms
14,320 KB
testcase_13 AC 114 ms
14,316 KB
testcase_14 AC 434 ms
23,024 KB
testcase_15 AC 380 ms
23,276 KB
testcase_16 AC 380 ms
23,020 KB
testcase_17 AC 382 ms
22,968 KB
testcase_18 AC 395 ms
23,020 KB
testcase_19 AC 250 ms
14,036 KB
testcase_20 AC 261 ms
13,912 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <queue>
#include <string>
#include <map>
#include <set>
#include <stack>
#include <tuple>
#include <deque>
#include <array>
#include <numeric>
#include <bitset>
#include <iomanip>
#include <cassert>
#include <chrono>
#include <random>
#include <limits>
#include <iterator>
#include <functional>
#include <sstream>
#include <fstream>
#include <complex>
#include <cstring>
#include <unordered_map>
#include <unordered_set>
using namespace std;

using ll = long long;
constexpr int INF = 1001001001;
constexpr int mod = 1000000007;
// constexpr int mod = 998244353;

template<class T>
inline bool chmax(T& x, T y){
    if(x < y){
        x = y;
        return true;
    }
    return false;
}
template<class T>
inline bool chmin(T& x, T y){
    if(x > y){
        x = y;
        return true;
    }
    return false;
}

struct UnionFind{
    int sz;
    vector<int> par;
    vector<int> rank;

    UnionFind(int n) : sz(n) {
        par.resize(sz);
        rank.assign(sz, 0);
        for(int i = 0; i < sz; ++i){
            par[i] = i;
        }
    }

    int find(int x){
        if(par[x] == x) return x;
        else    return par[x] = find(par[x]);
    }

    bool same(int x, int y){
        return find(x) == find(y);
    }

    bool unite(int x, int y){
        x = find(x), y = find(y);
        if(x == y)  return false;
        if(rank[x] < rank[y])   par[x] = y;
        else{
            par[y] = x;
            if(rank[x] == rank[y])  ++rank[x];
        }
        return true;
    }
};

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int N, M, Q;
    cin >> N >> M >> Q;
    vector<int> A(M), B(M), C(Q), D(Q);
    for(int i = 0; i < M; ++i){
        cin >> A[i] >> B[i];
        --A[i], --B[i];
    }
    set<pair<int, int>> destroy;
    for(int i = 0; i < Q; ++i){
        cin >> C[i] >> D[i];
        --C[i], --D[i];
        destroy.emplace(minmax(C[i], D[i]));
    }
    vector<bool> remain(M);
    for(int i = 0; i < M; ++i){
        remain[i] = destroy.find(minmax(A[i], B[i])) == destroy.end();
    }
    vector<int> lb(N, -1), ub(N, Q + 1);
    bool update = true;
    int cnt = 0;
    while(update){
        cnt += 1;
        assert(cnt <= 100);
        update = false;
        UnionFind uf(N);
        for(int i = 0; i < M; ++i){
            if(remain[i])   uf.unite(A[i], B[i]);
        }
        vector<vector<int>> querys(Q + 1);
        for(int i = 1; i < N; ++i){
            if(ub[i] - lb[i] > 1){
                querys[(lb[i] + ub[i]) / 2].emplace_back(i);
            }
        }
        for(int i : querys[Q]){
            if(uf.same(0, i))   lb[i] = Q;
            else    ub[i] = Q;
            update = true;
        }
        for(int q = Q - 1; q >= 0; --q){
            uf.unite(C[q], D[q]);
            for(int i : querys[q]){
                if(uf.same(0, i))   lb[i] = q;
                else    ub[i] = q;
                update = true;
            }
        }
    }
    // cout << "remain\n";
    // for(int i = 0; i < M; ++i){
    //     cout << remain[i] << ' ';
    // }cout << endl;
    // cout << "cnt:" << cnt << endl;
    // for(int i = 1; i < N; ++i){
    //     cout << lb[i] << ' ';
    // }cout << endl;
    for(int i = 1; i < N; ++i){
        cout << (lb[i] == Q ? -1 : lb[i] + 1) << '\n';
    }

    return 0;
}
0