結果

問題 No.1605 Matrix Shape
ユーザー nawawan
提出日時 2021-07-16 23:31:31
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 256 ms / 2,000 ms
コード長 1,515 bytes
コンパイル時間 1,629 ms
コンパイル使用メモリ 181,452 KB
実行使用メモリ 17,280 KB
最終ジャッジ日時 2024-07-06 11:08:59
合計ジャッジ時間 5,952 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
struct UnionFind{
    vector<int> par;
    vector<int> rank;
    UnionFind(int n){
        par.resize(n);
        rank.resize(n);
        for(int i = 0; i < n; i++){
            par[i] = i;
            rank[i] = 1;
        }
    }

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

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

    void unite(int x, int y){
        x = root(x);
        y = root(y);
        if(x == y) return;
        if(rank[x] < rank[y]) swap(x, y);
        par[y] = x;
        rank[x] += rank[y];
    }

    int size(int x) {
        return rank[root(x)];
    }
};
int main(){
    int N;
    cin >> N;
    vector<int> H(N), W(N);
    UnionFind U(200010);
    vector<int> iri(200010), de(200010);
    set<int> s;
    for(int i = 0; i < N; i++){
        cin >> H[i] >> W[i];
        H[i]--;
        W[i]--;
        s.insert(H[i]);
        s.insert(W[i]);
        U.unite(H[i], W[i]);
        iri[W[i]]++;
        de[H[i]]++;
    }
    int temp = 0;
    for(int i = 0; i < 200010; i++){
        temp = max(temp, U.size(i));
    }
    if(temp == (int)s.size()){
        map<int, int> m;
        for(int i = 0; i < 200010; i++){
            if(s.count(i)) m[iri[i] - de[i]]++;
        }
        if(m[0] == (int)s.size()) cout << s.size() << endl;
        else if(m[-1] == 1 && m[1] == 1) cout << 1 << endl;
        else cout << 0 << endl;
    }
    else cout << 0 << endl;
}
0