oj.uz

Submission #584845

# Submission time Handle Problem Language Result Execution time Memory
584845 2022-06-28T04:52:20 Z eecs Paint (COI20_paint) C++17
48 / 100
 988 ms 524288 KB 
#include <bits/stdc++.h>
using namespace std;

const int maxn = 200010, B = 500;
int r, c, q, bel[maxn], col[maxn];
vector<int> V[maxn], adj[maxn];
vector<unordered_set<int>> S[maxn];
bool mark[maxn];
set<int> big;

int main() {
    ios::sync_with_stdio(0), cin.tie(0);
    cin >> r >> c;
    for (int i = 0; i < r * c; i++) {
        cin >> col[i];
        bel[i] = i, V[i] = {i};
    }
    for (int i = 0; i < r; i++) {
        for (int j = 0; j < c; j++) {
            for (auto dir : {array<int, 2>{-1, 0}, {1, 0}, {0, -1}, {0, 1}}) {
                int x = i + dir[0], y = j + dir[1];
                if (!~x || x == r || !~y || y == c) continue;
                adj[i * c + j].push_back(x * c + y);
            }
        }
    }
    auto merge = [&](int u, int v) {
        if (V[u].size() < V[v].size()) swap(u, v);
        if (V[u].size() < B && V[u].size() + V[v].size() >= B) {
            S[u].resize(1e5), big.insert(u);
            for (int x : adj[u]) S[u][col[bel[x]]].insert(bel[x]);
        }
        if (V[u].size() + V[v].size() >= B) {
            for (int x : adj[v]) S[u][col[bel[x]]].insert(bel[x]);
            S[v].clear(), S[v].shrink_to_fit(), big.erase(v);
        }
        for (int x : V[v]) bel[x] = u;
        copy(V[v].begin(), V[v].end(), back_inserter(V[u]));
        copy(adj[v].begin(), adj[v].end(), back_inserter(adj[u]));
    };
    auto fill = [&](int u, int x) {
        u = bel[u];
        while (1) {
            if (S[u].size() < B) {
                bool flag = 0;
                auto tmp = adj[u];
                for (int v : tmp) {
                    v = bel[v];
                    if (u ^ v && col[v] == x) merge(u, v), flag = 1, u = bel[u];
                }
                if (!flag) break;
            } else {
                if (S[u][x].empty()) break;
                int v = *S[u][x].begin();
                S[u][x].erase(v), v = bel[v];
                if (u ^ v && col[v] == x) merge(u, v), u = bel[u];
            }
        }
        col[u] = x;
        if (S[u].size() < B) {
            vector<int> _adj;
            for (int v : adj[u]) {
                v = bel[v];
                if (u ^ v && !mark[v]) _adj.push_back(v), mark[v] = 1;
            }
            adj[u] = _adj;
            for (int v : big) {
                if (mark[v]) S[v][col[u]].insert(u);
            }
            for (int v : adj[u]) mark[v] = 0;
        } else {
            for (int v : big) {
                if (u ^ v && S[u][col[v]].count(v)) S[v][col[u]].insert(u);
            }
        }
    };
    for (int i = 0; i < r * c; i++) {
        fill(i, col[i]);
    }
    cin >> q;
    while (q--) {
        int i, j, x;
        cin >> i >> j >> x;
        fill(--i * c + --j, x);
    }
    for (int i = 0; i < r; i++) {
        for (int j = 0; j < c; j++) {
            cout << col[bel[i * c + j]] << " ";
        }
        cout << "\n";
    }
    return 0;
}

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 8 ms 14428 KB Output is correct
2 Correct 9 ms 14548 KB Output is correct
3 Correct 17 ms 15188 KB Output is correct
4 Correct 21 ms 20704 KB Output is correct
5 Correct 20 ms 21740 KB Output is correct
6 Correct 45 ms 31840 KB Output is correct
7 Correct 8 ms 14420 KB Output is correct

Subtask #2
9.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 62 ms 18780 KB Output is correct
2 Correct 304 ms 276644 KB Output is correct
3 Correct 147 ms 40956 KB Output is correct

Subtask #3
31.0 / 31.0

# Verdict Execution time Memory Grader output
1 Correct 184 ms 58668 KB Output is correct
2 Correct 185 ms 60080 KB Output is correct
3 Correct 234 ms 83104 KB Output is correct
4 Correct 331 ms 190112 KB Output is correct
5 Correct 285 ms 199768 KB Output is correct

Subtask #4
0 / 52.0

# Verdict Execution time Memory Grader output
1 Correct 181 ms 43764 KB Output is correct
2 Correct 353 ms 50904 KB Output is correct
3 Correct 988 ms 385000 KB Output is correct
4 Runtime error 328 ms 524288 KB Execution killed with signal 9
5 Halted 0 ms 0 KB -