結果

問題 No.157 2つの空洞
コンテスト
ユーザー tokizo
提出日時 2019-02-02 11:56:11
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 2,319 bytes
コンパイル時間 1,661 ms
コンパイル使用メモリ 174,224 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-27 08:23:56
合計ジャッジ時間 2,392 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 4
other WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

int h, w;
const int MAX = 30;
int dx[4] = {-1, 1, 0, 0};
int dy[4] = {0, 0, -1, 1};

char C[MAX][MAX];
bool seen[MAX][MAX];
int D[MAX][MAX];

struct UnionFind{
    vector<int> par;
    UnionFind(int N) : par(N){
        for(int i = 0; i < N; i++){
            par[i] = -1;
        }
    }
    int root(int x){
        if(par[x] < 0) return x;
        else return par[x] = root(par[x]);
    }
    void unite(int x, int y){
        int rx = root(x);
        int ry = root(y);
        if(rx == ry) return;
        else{
            if(par[rx] == par[ry]){
                if(rx > ry) swap(rx, ry);
            }else if(par[rx] > par[ry]) swap(rx, ry);
            par[rx] += par[ry];
            par[ry] = rx;
       }
    }
    bool same(int x, int y){
        return root(x) == root(y);
    }
    void check(){
        for(auto x : par){
            cout << x << " ";
        }
        cout << endl;
    }
};

void dfs(int x, int y, UnionFind &uf){
    seen[x][y] = true;
    for(int i = 0; i < 4; i++){
        int nx, ny;
        nx = dx[i] + x;
        ny = dy[i] + y;
        if(nx < 0 or ny < 0 or nx >= h or ny >= w) continue;
        if(C[nx][ny] == '#' or seen[nx][ny]) continue;
        uf.unite(x * w + y, nx * w + ny);
        dfs(nx, ny, uf);
    }
    return;
}

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

    cin >> w >> h;
    for(int i = 0; i < h; i++){
        for(int j = 0; j < w; j++){
            cin >> C[i][j];
            D[i][j] = 1e9;
        }
    }

    UnionFind uf(h * w);
    for(int i = 0; i < h; i++){
        for(int j = 0; j < w; j++){
            if(C[i][j] == '.'){
                if(!seen[i][j]){
                    dfs(i, j, uf);
                }
            }
        }
    }

    queue<pair<int, int>> q;
    bool f = true;
    int bp;
    for(int i = 1; i < h - 1; i++){
        for(int j = 1; j < w - 1; j++){
            int r = uf.root(i * w + j);
            if(f and uf.par[r] != -1){
                bp = r;
                f = false;
                q.push({i, j});
            }else if(!f and r == bp){
                q.push({i, j});
            }
        }
    }

    for(int i = 0; i < h * w; i++){
        cout << uf.par[i] << ' ';
        if((i + 1) % w == 0) cout << endl;
    }

    return 0;
}
0