結果

問題 No.307 最近色塗る問題多くない？
ユーザー んんん
提出日時 2023-01-05 23:38:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 3,585 bytes
コンパイル時間 2,687 ms
コンパイル使用メモリ 218,616 KB
最終ジャッジ日時 2025-02-09 23:26:00
ジャッジサーバーID
（参考情報） 		judge4 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 33 TLE * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

struct UnionFind {
    vector<int> data;

    UnionFind(int sz) {
        data.assign(sz, -1);
    }

    bool unite(int x, int y) {
        x = find(x), y = find(y);
        if (x == y) return false;
        data[y] = x;
        return true;
    }

    int find(int k) {
        if (data[k] < 0) return k;
        return data[k] = find(data[k]);
    }

    int color(int k) {
        return -data[find(k)]-1;
    }

    void change(int k, int a) {
        data[find(k)] = -a-1;
    }
};
int H, W;

int idx(int h, int w) {
    if (H > W) return h*H + w;
    return h + W*w;
}

int dx[4] = {0, 1, 0, -1};
int dy[4] = {1, 0, -1, 0};

int main() {
    cin >> H >> W;
    vector<vector<int>> A(H, vector<int>(W));
    for (auto &v : A) for (auto &e : v) cin >> e;

    UnionFind uf(max(H, W)*max(H, W));
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            int id = idx(i, j);
            uf.change(id, A[i][j]);
        }
    }

    vector<set<pair<int, int>>> vst(max(H, W)*max(H, W));
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            int id = idx(i, j);
            for (int d = 0; d < 4; d++) {
                int ni = i + dx[d];
                int nj = j + dy[d];
                if (ni < 0 || ni >= H || nj < 0 || nj >= W) continue;
                int nid = idx(ni, nj);
                if (uf.color(id) == uf.color(nid)) uf.unite(id, nid);
            }
        }
    }

    
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) {
            bool f = false;
            int id = idx(i, j);
            for (int d = 0; d < 4; d++) {
                int ni = i + dx[d];
                int nj = j + dy[d];
                if (ni < 0 || ni >= H || nj < 0 || nj >= W) continue;
                int nid = idx(ni, nj);
                if (uf.find(id) != uf.find(nid)) {
                    f = true;
                }
            }
            vst[uf.find(id)].emplace(i, j);
        }
    }

    int Q;
    cin >> Q;
    while (Q--) {
        int r, c, x;
        cin >> r >> c >> x;
        r--; c--;
        int sid = idx(r, c);
        int p = uf.find(sid);
        set<pair<int, int>> st;
        for (auto &[i, j] : vst[p]) {
            int id = idx(i, j);
            for (int d = 0; d < 4; d++) {
                int ni = i + dx[d];
                int nj = j + dy[d];
                if (ni < 0 || ni >= H || nj < 0 || nj >= W) continue;
                int nid = idx(ni, nj);
                if (uf.find(id) != uf.find(nid) && uf.color(nid) == x) {
                    int np = uf.find(nid);
                    st.insert(vst[np].begin(), vst[np].end());
                    vst[np].clear();
                    uf.unite(p, nid);
                }
            }
        }
        uf.change(sid, x);
        vst[p].insert(st.begin(), st.end());
        vector<pair<int, int>> vp;
        for (auto &[i, j] : vst[p]) {
            bool f = true;
            int id = idx(i, j);
            for (int d = 0; d < 4; d++) {
                int ni = i + dx[d];
                int nj = j + dy[d];
                if (ni < 0 || ni >= H || nj < 0 || nj >= W) continue;
                int nid = idx(ni, nj);
                if (uf.find(id) != uf.find(nid)) f = false;
            }
            if (f) vp.emplace_back(make_pair(i, j));
        }

        for (auto pa : vp) vst[p].erase(pa);
    }

    for (int i = 0; i < H; i++) for (int j = 0; j < W; j++) {
        int id = idx(i, j);
        cout << uf.color(id) << (j == W-1 ? "\n" : " ");
    }
}
0