結果

問題 No.2946 Puyo
ユーザー liveworldlikeliveworldlike
提出日時 2024-10-25 21:53:55
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 35 ms / 2,000 ms
コード長 1,708 bytes
コンパイル時間 572 ms
コンパイル使用メモリ 57,168 KB
実行使用メモリ 19,868 KB
最終ジャッジ日時 2024-10-25 21:54:03
合計ジャッジ時間 3,725 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff # 

#pragma GCC optimize("O3")
#include <map>
#include <stdio.h>
#include <stdlib.h>
using namespace std;
typedef __int64_t ll;

class DSU {
public:
  ll *size;
  ll *parent;
  DSU(ll n) {
    size = (ll *)calloc(n, sizeof(ll));
    parent = (ll *)calloc(n, sizeof(ll));
  }
  void add(ll k) {
    parent[k] = k;
    size[k] = 1;
  }
  ll get(ll k) {
    if (!size[k]) {
      add(k);
    }
    if (parent[k] == k) {
      return k;
    }
    return parent[k] = get(parent[k]);
  }
  ll query(ll k) {
    k = get(k);
    return size[k];
  }
  void merge(ll a, ll b) {
    a = get(a);
    b = get(b);
    if (a == b) {
      return;
    }
    if (size[a] < size[b]) {
      swap(a, b);
    }
    parent[b] = a;
    size[a] += size[b];
  }
};

void solve() {
  ll h, w;
  scanf("%ld %ld\n", &h, &w);
  auto cvt = [&](ll i, ll j) { return i * w + j; };
  char **g = (char **)malloc(h * sizeof(char *));
  for (ll i = 0; i < h; i++) {
    g[i] = (char *)malloc(w * sizeof(char));
    for (ll j = 0; j < w; j++) {
      g[i][j] = getc(stdin);
    }
    getc(stdin);
  }
  DSU dsu(h * w);
  for (ll i = 0; i < h - 1; i++) {
    for (ll j = 0; j < w; j++) {
      char a = g[i][j], b = g[i + 1][j];
      if (a == b) {
        dsu.merge(cvt(i, j), cvt(i + 1, j));
      }
    }
  }
  for (ll i = 0; i < h; i++) {
    for (ll j = 0; j < w - 1; j++) {
      char a = g[i][j], b = g[i][j + 1];
      if (a == b) {
        dsu.merge(cvt(i, j), cvt(i, j + 1));
      }
    }
  }
  for (ll i = 0; i < h; i++) {
    for (ll j = 0; j < w; j++) {
      if (dsu.query(cvt(i, j)) >= 4) {
        putc('.', stdout);
      } else {
        putc(g[i][j], stdout);
      }
    }
    putc('\n', stdout);
  }
}

int main() { solve(); }
0