結果

問題 No.2946 Puyo
ユーザー GOTKAKOGOTKAKO
提出日時 2024-10-25 21:36:57
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 1,351 bytes
コンパイル時間 2,352 ms
コンパイル使用メモリ 213,604 KB
実行使用メモリ 12,228 KB
最終ジャッジ日時 2024-10-25 21:37:06
合計ジャッジ時間 5,529 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff #

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

class UnionFind{
    public:
    vector<int> par,siz;
 
    void make(int N){
        par.resize(N,-1);
        siz.resize(N,1);
    }
 
    int root(int x){
        if(par.at(x) == -1) return x;
        else return par.at(x) = root(par.at(x));
    }
 
    void unite(int u, int v){
        u = root(u),v = root(v);
        if(u == v) return;
        if(siz.at(u) < siz.at(v)) swap(u,v);
 
        par.at(v) = u;
        siz.at(u) += siz.at(v);
    }
 
    bool issame(int u, int v){
        if(root(u) == root(v)) return true;
        else return false;
    }
};

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int H,W; cin >> H >> W;
    vector<string> S(H);
    for(auto &s : S) cin >> s;
    int n = H*W;
    UnionFind Z; Z.make(n);

    vector<pair<int,int>> dxy = {{-1,0},{0,1},{1,0},{0,-1}};
    for(int i=0; i<H; i++) for(int k=0; k<W; k++){
        int pos = i*W+k;
        for(auto &[dx,dy] : dxy){
            int nx = i+dx,ny = k+dy;
            if(nx < 0 || ny < 0 || nx >= H || ny >= W) continue;
            int pos2 = nx*W+ny;
            if(S.at(i).at(k) == S.at(nx).at(ny)) Z.unite(pos,pos2);
        }
    }
    for(int i=0; i<H; i++) for(int k=0; k<W; k++) if(Z.siz.at(Z.root(i*W+k)) >= 4) S.at(i).at(k) = '.';
    for(auto &s : S) cout << s << "\n";
} 
0