結果

問題 No.697 池の数はいくつか
ユーザー rieaaddlreiuu
提出日時 2024-05-26 21:05:49
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,810 ms / 6,000 ms
コード長 2,516 bytes
コンパイル時間 2,305 ms
コンパイル使用メモリ 180,016 KB
実行使用メモリ 214,668 KB
最終ジャッジ日時 2024-12-20 20:25:29
合計ジャッジ時間 24,228 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

struct UnionFind {
    vector<int> par, rank, siz;
    UnionFind(int n) : par(n,-1), rank(n,0), siz(n,1) { }
    int root(int x) {
        if (par[x]==-1) return x; // x が根の場合は x を返す
        else return par[x] = root(par[x]); // 経路圧縮
    }
    bool issame(int x, int y) {
        return root(x)==root(y);
    }
    bool unite(int x, int y) {
        int rx = root(x), ry = root(y); // x 側と y 側の根を取得する
        if (rx==ry) return false; // すでに同じグループのときは何もしない
        // union by rank
        if (rank[rx]<rank[ry]) swap(rx, ry); // ry 側の rank が小さくなるようにする
        par[ry] = rx; // ry を rx の子とする
        if (rank[rx]==rank[ry]) rank[rx]++; // rx 側の rank を調整する
        siz[rx] += siz[ry]; // rx 側の siz を調整する
        return true;
    }
    int size(int x) {
        return siz[root(x)];
    }
};
void array_distinct(vector<int> &a,vector<int> &result){
    for(int i=0;i<a.size();i++){
        if(i == a.size() - 1){
            result.push_back(a[i]);
        } else if(a[i] != a[i+1]){
            result.push_back(a[i]);
        }
    }
}

void vector_output(vector<int> &v){
    cout << "{ ";
    for(int i=0;i<v.size();i++){
        cout << v[i] << " ";
    }
    cout << "}" << endl;
}

int main(){
    int H,W;
    cin >> H >> W;
    vector<vector<int>> grid(W+2,vector<int>(H+2));
    int u_size = (H+2)*(W+2); 
    UnionFind u(u_size);
    for(int i=0;i<W+2;i++){
        grid[i][0] = 0;
        grid[i][H+1] = 0;
    }
    for(int i=0;i<H+2;i++){
        grid[0][i] = 0;
        grid[W+1][i] = 0;
    }
    for(int j=1;j<H+1;j++){
        for(int i=1;i<W+1;i++){
            cin >> grid[i][j];
        }
    }
    for(int j=1;j<H+1;j++){
        for(int i=1;i<W+1;i++){
            if(grid[i][j] == grid[i][j-1]){
                u.unite(j*(W+2)+i,(j-1)*(W+2)+i);
            }
            if(grid[i][j] == grid[i-1][j]){
                u.unite(j*(W+2)+i,j*(W+2)+i-1);
            }
        }
    }
    int grid_i,grid_j,count = 0;
    vector<int> roots(u_size);
    vector<int> roots_distinct;
    for(int i=0;i<u_size;i++){
        grid_i = u.root(i)% (W+2);
        grid_j = u.root(i)/ (W+2);
        if(grid[grid_i][grid_j] == 1){
            roots.push_back(u.root(i));
        }
    }
    sort(roots.begin(),roots.end());
    array_distinct(roots,roots_distinct);
    cout << roots_distinct.size() - 1<< endl;
}
0