結果

問題 No.416 旅行会社
ユーザー outlineoutline
提出日時 2021-01-30 18:04:29
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 443 ms / 4,000 ms
コード長 3,068 bytes
コンパイル時間 1,702 ms
コンパイル使用メモリ 150,748 KB
実行使用メモリ 23,172 KB
最終ジャッジ日時 2023-08-21 10:44:35
合計ジャッジ時間 6,841 ms
ジャッジサーバーID
(参考情報)
judge13 / judge14
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 122 ms
14,284 KB
testcase_01 AC 1 ms
4,376 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 2 ms
4,380 KB
testcase_06 AC 2 ms
4,376 KB
testcase_07 AC 3 ms
4,376 KB
testcase_08 AC 5 ms
4,376 KB
testcase_09 AC 20 ms
4,688 KB
testcase_10 AC 144 ms
14,292 KB
testcase_11 AC 132 ms
14,064 KB
testcase_12 AC 138 ms
14,056 KB
testcase_13 AC 109 ms
14,028 KB
testcase_14 AC 443 ms
22,752 KB
testcase_15 AC 412 ms
23,172 KB
testcase_16 AC 409 ms
22,908 KB
testcase_17 AC 430 ms
22,904 KB
testcase_18 AC 431 ms
22,892 KB
testcase_19 AC 265 ms
13,900 KB
testcase_20 AC 277 ms
14,024 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;
    while(update){
        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;
            }
        }
    }
    
    for(int i = 1; i < N; ++i){
        cout << (lb[i] == Q ? -1 : lb[i] + 1) << '\n';
    }

    return 0;
}
0