#include <bits/stdc++.h>
using namespace std;
struct DSU {
vector<int> root, col, sz;
void init(int n) {
root.assign(n, 0);
col.assign(n, 0);
sz.assign(n, 1);
iota(root.begin(), root.end(), 0);
}
int findRoot(int u) {
if (u == root[u])
return u;
return root[u] = findRoot(root[u]);
}
void SetCol(int u, int c) {
col[findRoot(u)] = c;
}
int GetCol(int u) {
return col[findRoot(u)];
}
int GetSz(int u) {
return sz[findRoot(u)];
}
void unite(int u, int v) {
u = findRoot(u);
v = findRoot(v);
assert(col[u] == col[v]);
if (u == v)
return;
if (sz[u] < sz[v])
swap(u, v);
root[v] = u;
sz[u] += sz[v];
}
};
const int dx[4] = {0, 0, -1, 1};
const int dy[4] = {-1, 1, 0, 0};
const int BLOCKSIZE = 2222;
int m, n, q, curInd;
vector<vector<int>> a, ind;
vector<set<int>> adj;
vector<set<pair<int, int>>> s;
DSU dsu;
void Update(int u, int type) {
assert(dsu.GetSz(u) < BLOCKSIZE);
u = dsu.findRoot(u);
set<int> newAdj;
for (int v : adj[u]) {
v = dsu.findRoot(v);
newAdj.insert(v);
if (dsu.GetSz(v) < BLOCKSIZE)
continue;
if (type == 1) {
s[v].insert({dsu.GetCol(u), u});
} else {
s[v].erase({dsu.GetCol(u), u});
}
}
adj[u] = newAdj;
}
void Merge(int u, int v) {
u = dsu.findRoot(u);
v = dsu.findRoot(v);
if (u == v)
return;
if (dsu.GetSz(u) < dsu.GetSz(v))
swap(u, v);
if (dsu.GetSz(u) < BLOCKSIZE)
Update(u, -1);
if (dsu.GetSz(v) < BLOCKSIZE)
Update(v, -1);
for (int x : adj[v]) {
adj[u].insert(x);
if (dsu.GetSz(u) >= BLOCKSIZE && dsu.GetSz(x) < BLOCKSIZE)
s[u].insert({dsu.GetCol(x), dsu.findRoot(x)});
}
for (auto p : s[v])
s[u].insert(p);
s[v].clear();
adj[v].clear();
int preSz = dsu.GetSz(u);
dsu.unite(u, v);
adj[v].clear();
if (preSz < BLOCKSIZE && dsu.GetSz(u) >= BLOCKSIZE) {
set<int> newAdj;
for (int x : adj[u])
if (dsu.GetSz(x) >= BLOCKSIZE)
newAdj.insert(dsu.findRoot(x));
else
s[u].insert({dsu.GetCol(x), dsu.findRoot(x)});
adj[u] = newAdj;
} else if (dsu.GetSz(u) < BLOCKSIZE) {
Update(u, 1);
} else if (dsu.GetSz(u) >= BLOCKSIZE) {
set<int> newAdj;
for (int x : adj[u])
if (dsu.GetSz(x) >= BLOCKSIZE)
newAdj.insert(dsu.findRoot(x));
adj[u] = newAdj;
}
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cin >> m >> n;
a.assign(m, vector<int>(n, 0));
ind.assign(m, vector<int>(n, 0));
adj.assign(m * n, set<int>());
s.assign(m * n, set<pair<int, int>>());
dsu.init(m * n);
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++) {
cin >> a[i][j];
ind[i][j] = curInd++;
dsu.SetCol(ind[i][j], a[i][j]);
}
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++)
for (int dir = 0; dir < 4; dir++)
if (0 <= i + dx[dir] && i + dx[dir] < m && 0 <= j + dy[dir] && j + dy[dir] < n && a[i][j] == a[i + dx[dir]][j + dy[dir]])
dsu.unite(ind[i][j], ind[i + dx[dir]][j + dy[dir]]);
for (int i = 0; i < m; i++)
for (int j = 0; j < n; j++)
for (int dir = 0; dir < 4; dir++)
if (0 <= i + dx[dir] && i + dx[dir] < m && 0 <= j + dy[dir] && j + dy[dir] < n && a[i][j] != a[i + dx[dir]][j + dy[dir]]) {
int u = dsu.findRoot(ind[i][j]), v = dsu.findRoot(ind[i + dx[dir]][j + dy[dir]]);
if (dsu.GetSz(u) < BLOCKSIZE || dsu.GetSz(v) >= BLOCKSIZE)
adj[u].insert(v);
if (dsu.GetSz(v) < BLOCKSIZE || dsu.GetSz(u) >= BLOCKSIZE)
adj[v].insert(u);
}
for (int u = 0; u < m * n; u++)
if (dsu.findRoot(u) == u)
if (dsu.GetSz(u) < BLOCKSIZE)
Update(u, 1);
cin >> q;
while (q--) {
int x, y, c;
cin >> x >> y >> c;
x--, y--;
int u = dsu.findRoot(ind[x][y]);
if (dsu.GetSz(u) < BLOCKSIZE)
Update(u, -1);
dsu.SetCol(u, c);
if (dsu.GetSz(u) < BLOCKSIZE)
Update(u, 1);
vector<int> shouldMerge;
vector<pair<int, int>> shouldDel;
auto it = s[u].lower_bound({c, 0});
while (it != s[u].end())
if ((*it).first == c) {
shouldMerge.push_back((*it).second);
shouldDel.push_back((*it));
++it;
} else {
break;
}
for(auto p : shouldDel)
s[u].erase(p);
for (int v : adj[u])
if (dsu.GetCol(v) == c)
shouldMerge.push_back(v);
for (int v : shouldMerge) {
Merge(u, v);
u = dsu.findRoot(u);
}
}
for (int i = 0; i < m; i++) {
for (int j = 0; j < n; j++)
cout << dsu.GetCol(ind[i][j]) << ' ';
cout << '\n';
}
return 0;
}
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
2 ms |
460 KB |
Output is correct |
| 2 |
Correct |
2 ms |
588 KB |
Output is correct |
| 3 |
Correct |
14 ms |
3320 KB |
Output is correct |
| 4 |
Correct |
55 ms |
2676 KB |
Output is correct |
| 5 |
Correct |
18 ms |
2440 KB |
Output is correct |
| 6 |
Correct |
64 ms |
2420 KB |
Output is correct |
| 7 |
Correct |
1 ms |
204 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
161 ms |
10352 KB |
Output is correct |
| 2 |
Correct |
221 ms |
18228 KB |
Output is correct |
| 3 |
Correct |
285 ms |
34808 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
664 ms |
60252 KB |
Output is correct |
| 2 |
Correct |
747 ms |
60104 KB |
Output is correct |
| 3 |
Correct |
980 ms |
59688 KB |
Output is correct |
| 4 |
Correct |
1134 ms |
57420 KB |
Output is correct |
| 5 |
Correct |
652 ms |
54252 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
724 ms |
41300 KB |
Output is correct |
| 2 |
Correct |
664 ms |
43164 KB |
Output is correct |
| 3 |
Correct |
2539 ms |
42752 KB |
Output is correct |
| 4 |
Execution timed out |
3056 ms |
34952 KB |
Time limit exceeded |
| 5 |
Halted |
0 ms |
0 KB |
- |
