#1032146 (C++23) No.2946 Puyo

提出ソース

結果

問題	No.2946 Puyo
ユーザー	Slephy
提出日時	2024-12-18 21:53:44
言語	C++23 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	51 ms / 2,000 ms
コード長	5,324 bytes
コンパイル時間	3,214 ms
コンパイル使用メモリ	257,268 KB
実行使用メモリ	8,200 KB
最終ジャッジ日時	2024-12-18 21:53:54
合計ジャッジ時間	9,214 ms
ジャッジサーバーID （参考情報）	judge4 / judge2

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,820 KB
testcase_01	AC	2 ms 6,816 KB
testcase_02	AC	2 ms 6,816 KB
testcase_03	AC	46 ms 8,144 KB
testcase_04	AC	51 ms 8,200 KB
testcase_05	AC	47 ms 8,100 KB
testcase_06	AC	51 ms 8,156 KB
testcase_07	AC	47 ms 8,112 KB
testcase_08	AC	5 ms 6,820 KB
testcase_09	AC	17 ms 6,820 KB
testcase_10	AC	4 ms 6,820 KB
testcase_11	AC	25 ms 6,816 KB
testcase_12	AC	4 ms 6,816 KB
testcase_13	AC	2 ms 6,820 KB
testcase_14	AC	7 ms 6,816 KB
testcase_15	AC	5 ms 6,816 KB
testcase_16	AC	10 ms 6,820 KB
testcase_17	AC	30 ms 6,820 KB
testcase_18	AC	3 ms 6,816 KB
testcase_19	AC	16 ms 6,820 KB
testcase_20	AC	20 ms 6,816 KB
testcase_21	AC	32 ms 6,816 KB
testcase_22	AC	6 ms 6,820 KB
testcase_23	AC	20 ms 6,820 KB
testcase_24	AC	46 ms 6,816 KB
testcase_25	AC	40 ms 6,816 KB
testcase_26	AC	43 ms 6,820 KB
testcase_27	AC	2 ms 6,816 KB
testcase_28	AC	30 ms 6,816 KB
testcase_29	AC	41 ms 6,816 KB
testcase_30	AC	25 ms 6,816 KB
testcase_31	AC	35 ms 6,816 KB
testcase_32	AC	38 ms 6,816 KB
testcase_33	AC	31 ms 6,820 KB
testcase_34	AC	50 ms 6,816 KB
testcase_35	AC	36 ms 6,816 KB
testcase_36	AC	38 ms 6,820 KB
testcase_37	AC	40 ms 6,816 KB
testcase_38	AC	28 ms 6,820 KB
testcase_39	AC	26 ms 6,820 KB
testcase_40	AC	18 ms 6,820 KB
testcase_41	AC	35 ms 6,820 KB
testcase_42	AC	30 ms 6,820 KB
testcase_43	AC	28 ms 6,816 KB
testcase_44	AC	35 ms 6,820 KB
testcase_45	AC	30 ms 6,816 KB
testcase_46	AC	18 ms 6,820 KB
testcase_47	AC	29 ms 6,820 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int INF = (int)1e9 + 1001010;
const ll llINF = (long long)4e18 + 22000020;
const string endn = "\n";
template <class T> inline auto vector2(size_t i, size_t j, const T &init = T()) {
    return vector(i, vector<T>(j, init));
}
const string ELM_SEP = "", VEC_SEP = endn;
template <class T> istream &operator>>(istream &i, vector<T> &A) {
    for(auto &I : A) i >> I;
    return i;
}
template <class T> ostream &operator<<(ostream &o, const vector<T> &A) {
    int i = A.size();
    for(const auto &I : A) o << I << (--i ? ELM_SEP : "");
    return o;
}
template <class T> ostream &operator<<(ostream &o, const vector<vector<T>> &A) {
    int i = A.size();
    for(const auto &I : A) o << I << (--i ? VEC_SEP : "");
    return o;
}
template <class T> vector<T> &operator++(vector<T> &A, int) {
    for(auto &I : A) I++;
    return A;
}
template <class T> vector<T> &operator--(vector<T> &A, int) {
    for(auto &I : A) I--;
    return A;
}
template <class T, class U> bool chmax(T &a, const U &b) { return ((a < b) ? (a = b, true) : false); }
template <class T, class U> bool chmin(T &a, const U &b) { return ((a > b) ? (a = b, true) : false); }
ll floor(ll a, ll b) {
    if(b < 0) a = -a, b = -b;
    return (a >= 0) ? a / b : (a + 1) / b - 1;
}
ll ceil(ll a, ll b) {
    if(b < 0) a = -a, b = -b;
    return (a > 0) ? (a - 1) / b + 1 : a / b;
}
ll check_bit(unsigned long long val, unsigned long long digit) { return (val >> digit) & 1; }
#ifdef DEBUG
#include <cpp-dump/cpp-dump.hpp>
#define dump(...) cpp_dump(__VA_ARGS__)
namespace cp = cpp_dump;
struct InitCppDump {
    InitCppDump() {
        if(!isatty(fileno(stderr))) CPP_DUMP_SET_OPTION(es_style, cpp_dump::types::es_style_t::no_es);
        CPP_DUMP_SET_OPTION(log_label_func, cp::log_label::line());
        CPP_DUMP_SET_OPTION(max_iteration_count, 30);
    }
} init_cpp_dump;
#else
#define dump(...)
#endif
// ==================== ここまでテンプレ ====================

// (path compression) + (union by size)
// 参考 AtCoder Library
// == ACLからの変更点 ==
// group_num() を追加
// size関数をsize_of_group関数に改名
// groups() の返り値が昇順にソート済みであることを保証
class DSU{
    int n;
    int _group_num;
    vector<int> parent_or_size;

public:
    explicit DSU(size_t n) : n(n), _group_num(n), parent_or_size(n, -1) {}

    // 頂点u, vを結合し、それらの新たなリーダーを返す
    // O(α(N))
    int merge(int u, int v){
        assert(0 <= u && u < n);
        assert(0 <= v && v < n);
        u = leader(u);
        v = leader(v);
        if(u == v) return u;
        if(-parent_or_size[u] < -parent_or_size[v]) swap(u, v); // make size(u) >= size(v)
        parent_or_size[u] += parent_or_size[v];
        parent_or_size[v] = u;
        _group_num--;
        return u;
    }

    // 頂点u, vが同じ連結成分に属するかを判定する
    // O(α(N))
    bool same(int u, int v){
        assert(0 <= u && u < n);
        assert(0 <= v && v < n);
        return leader(u) == leader(v);
    }

    // 頂点vが属する連結成分の代表を返す
    // O(α(N))
    int leader(int v){
        assert(0 <= v && v < n);
        if(parent_or_size[v] < 0) return v;
        else return parent_or_size[v] = leader(parent_or_size[v]);
    }

    // 頂点vが属する連結成分のサイズを返す
    // O(α(N))
    int size_of_group(int v){
        assert(0 <= v && v < n);
        return -parent_or_size[leader(v)];
    }

    // 連結成分の個数を返す
    // 返り値は groups().size()に等しい
    // O(1)
    int group_num(){
        return _group_num;
    }

    // 連結成分ごとの頂点情報を返す
    // 1次元目、2次元目ともに昇順でソート済み
    // O(N)
    vector<vector<int>> groups(){
        vector<vector<int>> result;
        // 頂点iをリーダーとする連結成分がresultに格納されるときのインデックス
        vector<int> resultIndex(n, -1);

        for(int i = 0; i < n; i++){
            int ld = leader(i);
            if(resultIndex[ld] == -1){ // resultIndex が未定のとき
                resultIndex[ld] = result.size();
                result.emplace_back();
                result.back().reserve(size_of_group(ld));
            }
            result[resultIndex[ld]].emplace_back(i);
        }
        return result;
    }
};

int main(int argc, char *argv[]){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    ll h, w; cin >> h >> w;
    auto g = vector2<char>(h, w); cin >> g;

    DSU dsu(h * w);
    auto idx = [&](ll i, ll j) -> ll { return i * w + j; };

    // merge
    for(int i = 0; i < h; i++){
        for(int j = 0; j < w; j++){
            if(i < h-1){
                if(g[i][j] == g[i+1][j]){
                    dsu.merge(idx(i, j), idx(i+1, j));
                }
            }
            if(j < w-1){
                if(g[i][j] == g[i][j+1]){
                    dsu.merge(idx(i, j), idx(i, j+1));
                }
            }
        }
    }

    for(int i = 0; i < h; i++){
        for(int j = 0; j < w; j++){
            if(dsu.size_of_group(idx(i, j)) >= 4){
                g[i][j] = '.';
            }
        }
    }

    cout << g << endl;
    return 0;
}

yukicoder

結果

テストケース

ソースコード