結果

問題 No.307 最近色塗る問題多くない？
ユーザー んんんんんん
提出日時 2023-01-05 23:38:42
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 3,585 bytes
コンパイル時間 2,791 ms
コンパイル使用メモリ 229,152 KB
実行使用メモリ 14,116 KB
最終ジャッジ日時 2024-11-29 15:50:33
合計ジャッジ時間 23,142 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
10,496 KB
testcase_01 AC 2 ms
12,672 KB
testcase_02 AC 2 ms
10,624 KB
testcase_03 AC 2 ms
6,820 KB
testcase_04 AC 4 ms
6,816 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 12 ms
5,248 KB
testcase_08 AC 61 ms
7,040 KB
testcase_09 AC 26 ms
5,376 KB
testcase_10 AC 168 ms
5,248 KB
testcase_11 AC 1,176 ms
5,888 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 24 ms
5,248 KB
testcase_14 AC 5 ms
5,248 KB
testcase_15 AC 7 ms
5,248 KB
testcase_16 AC 6 ms
5,248 KB
testcase_17 AC 21 ms
5,248 KB
testcase_18 AC 35 ms
6,656 KB
testcase_19 AC 40 ms
7,040 KB
testcase_20 AC 4 ms
5,248 KB
testcase_21 AC 33 ms
7,168 KB
testcase_22 AC 40 ms
7,552 KB
testcase_23 AC 32 ms
7,040 KB
testcase_24 AC 37 ms
7,168 KB
testcase_25 AC 60 ms
6,912 KB
testcase_26 AC 61 ms
7,168 KB
testcase_27 AC 637 ms
7,296 KB
testcase_28 TLE -
testcase_29 AC 18 ms
5,248 KB
testcase_30 AC 70 ms
7,296 KB
testcase_31 AC 1,006 ms
7,424 KB
testcase_32 AC 57 ms
7,296 KB
testcase_33 AC 33 ms
8,960 KB
testcase_34 TLE -
testcase_35 TLE -
権限があれば一括ダウンロードができます

ソースコード

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