結果

問題 No.697 池の数はいくつか
ユーザー rieaaddlreiuurieaaddlreiuu
提出日時 2024-05-26 21:05:49
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 2,802 ms / 6,000 ms
コード長 2,516 bytes
コンパイル時間 2,683 ms
コンパイル使用メモリ 180,696 KB
実行使用メモリ 214,696 KB
最終ジャッジ日時 2024-05-26 21:06:16
合計ジャッジ時間 24,396 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 2 ms
6,944 KB
testcase_10 AC 2 ms
6,944 KB
testcase_11 AC 2 ms
6,944 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 2 ms
6,940 KB
testcase_14 AC 2 ms
6,944 KB
testcase_15 AC 2 ms
6,944 KB
testcase_16 AC 2 ms
6,940 KB
testcase_17 AC 2 ms
6,940 KB
testcase_18 AC 2 ms
6,940 KB
testcase_19 AC 2 ms
6,940 KB
testcase_20 AC 2 ms
6,948 KB
testcase_21 AC 2 ms
6,940 KB
testcase_22 AC 2 ms
6,944 KB
testcase_23 AC 2 ms
6,944 KB
testcase_24 AC 273 ms
27,900 KB
testcase_25 AC 286 ms
27,316 KB
testcase_26 AC 270 ms
27,368 KB
testcase_27 AC 267 ms
27,716 KB
testcase_28 AC 269 ms
28,412 KB
testcase_29 AC 2,530 ms
214,564 KB
testcase_30 AC 2,763 ms
214,580 KB
testcase_31 AC 2,442 ms
214,600 KB
testcase_32 AC 2,790 ms
214,560 KB
testcase_33 AC 2,776 ms
214,696 KB
testcase_34 AC 2,802 ms
214,612 KB
権限があれば一括ダウンロードができます

ソースコード

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