#include <cstdio>
#include <map>
#include <unordered_map>
using std::map;
using std::unordered_map;
const int NN = 100100;
const int ROWS = 2;
int data[NN][ROWS];
typedef int Color;
class UnionSet {
UnionSet *parent;
public:
UnionSet():parent(this){}
void init() { parent = this; }
bool isDisjoint(UnionSet *other) { return repr() != other->repr(); }
void joint(UnionSet *other) { other->repr()->parent = this; }
private:
UnionSet *repr() {
if (parent == this) {
return this;
} else {
return parent = parent->repr();
}
}
};
struct Adj {
Color a[3];
};
unordered_map<Color, Adj> *compute_adj(int Cols) {
unordered_map<Color, Adj> *ret = new unordered_map<int, Adj>;
for (int r = 0; r < ROWS; ++r) {
for (int c = 0; c < Cols; ++c) {
Adj x = {.a = {data[c][ROWS - r - 1], data[(c + 1) % Cols][r], data[(c + Cols - 1) % Cols][r]}};
ret->insert({data[c][r], x});
}
}
return ret;
}
map<int, int> compute(int colors, int K, unordered_map<Color, Adj> *data) {
map<int, int> res;
for (int i = 1; i <= K; ++i) {
res[i] = 0;
}
unordered_map<Color, UnionSet *> sets;
int count;
for (Color l = 1; l <= colors; ++l) {
for (auto p : sets) {
delete p.second;
}
sets.clear();
count = 0;
for (Color r = l; r <= colors; ++r) {
UnionSet *cur = new UnionSet();
sets[r] = cur;
count++;
for (Color c : (*data)[r].a) {
auto res = sets.find(c);
if (res != sets.end()) {
if (cur->isDisjoint(res->second)) {
cur->joint(res->second);
--count;
}
}
}
if (count <= K) {
res[count] += 1;
}
}
}
return res;
}
int main() {
int COLS, K;
scanf("%d%d", &COLS, &K);
for (int r = 0; r < ROWS; ++r) {
for (int c = 0; c < COLS; ++c) {
scanf("%d", &data[c][r]);
}
}
unordered_map<int, Adj> *adj = compute_adj(COLS);
map<int, int> res = compute(2 * COLS, K, adj);
for (auto p : res) {
printf("%d ", p.second);
}
printf("\n");
delete adj;
return 0;
}
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | #include <cstdio> #include <map> #include <unordered_map> using std::map; using std::unordered_map; const int NN = 100100; const int ROWS = 2; int data[NN][ROWS]; typedef int Color; class UnionSet { UnionSet *parent; public: UnionSet():parent(this){} void init() { parent = this; } bool isDisjoint(UnionSet *other) { return repr() != other->repr(); } void joint(UnionSet *other) { other->repr()->parent = this; } private: UnionSet *repr() { if (parent == this) { return this; } else { return parent = parent->repr(); } } }; struct Adj { Color a[3]; }; unordered_map<Color, Adj> *compute_adj(int Cols) { unordered_map<Color, Adj> *ret = new unordered_map<int, Adj>; for (int r = 0; r < ROWS; ++r) { for (int c = 0; c < Cols; ++c) { Adj x = {.a = {data[c][ROWS - r - 1], data[(c + 1) % Cols][r], data[(c + Cols - 1) % Cols][r]}}; ret->insert({data[c][r], x}); } } return ret; } map<int, int> compute(int colors, int K, unordered_map<Color, Adj> *data) { map<int, int> res; for (int i = 1; i <= K; ++i) { res[i] = 0; } unordered_map<Color, UnionSet *> sets; int count; for (Color l = 1; l <= colors; ++l) { for (auto p : sets) { delete p.second; } sets.clear(); count = 0; for (Color r = l; r <= colors; ++r) { UnionSet *cur = new UnionSet(); sets[r] = cur; count++; for (Color c : (*data)[r].a) { auto res = sets.find(c); if (res != sets.end()) { if (cur->isDisjoint(res->second)) { cur->joint(res->second); --count; } } } if (count <= K) { res[count] += 1; } } } return res; } int main() { int COLS, K; scanf("%d%d", &COLS, &K); for (int r = 0; r < ROWS; ++r) { for (int c = 0; c < COLS; ++c) { scanf("%d", &data[c][r]); } } unordered_map<int, Adj> *adj = compute_adj(COLS); map<int, int> res = compute(2 * COLS, K, adj); for (auto p : res) { printf("%d ", p.second); } printf("\n"); delete adj; return 0; } |
English