#include<cstdio> #include<iostream> #include<iomanip> #include<algorithm> #include<string> #include<vector> #include<cmath> #include<set> #include<unordered_map> using namespace std; typedef vector<int> VI; typedef long long LL; typedef vector<LL> VLL; typedef pair<int, int> PII; typedef vector<PII> VPII; #define FOR(x, b, e) for(int x=b; x<=(e); ++x) #define FORD(x, b, e) for(int x=b; x>=(e); --x) #define REP(x, n) for(int x=0; x<(n); ++x) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define PB push_back #define ST first #define ND second class UnionFind { private: unordered_map<int, int> id, rank, setSize; int i_size = 0; public: UnionFind() {} ~UnionFind() { id.clear(); rank.clear(); setSize.clear(); } int findSet(int i) { return (id[i] == i) ? i : (id[i] = findSet(id[i])); } bool connected(int i, int j) { return findSet(i) == findSet(j); } int size() { return i_size; } int sizeOfSet(int i) { return setSize[findSet(i)]; } void makeUnion(int i, int j) { if (connected(i, j)) return; int x = findSet(i), y = findSet(j); if (rank[x] > rank[y]) swap(x, y); id[x] = y; if (rank[x] == rank[y]) ++rank[y]; setSize[y] += setSize[x]; --i_size; } bool contains(int i) { return id.find(i) != id.end(); } void add(int i) { rank[i] = 0; id[i] = i; i_size++; } }; int main() { ios_base::sync_with_stdio(0); cin.tie(NULL); cout.tie(NULL); int n, k; cin >> n >> k; vector<VI> canvas(2, VI(n)); VPII positions(2 * n + 1); REP(i, n) { cin >> canvas[0][i]; positions[canvas[0][i]] = { 0, i }; } REP(i, n) { cin >> canvas[1][i]; positions[canvas[1][i]] = { 1, i }; } VI result(k); FOR(i, 1, 2 * n) { result[0]++; UnionFind uf; uf.add(i); FOR(j, i + 1, 2 * n) { uf.add(j); int left = canvas[positions[j].ST][(n + positions[j].ND - 1) % n]; int right = canvas[positions[j].ST][(positions[j].ND + 1) % n]; int vert = canvas[(positions[j].ST + 1) % 2][positions[j].ND]; if (uf.contains(left)) uf.makeUnion(j, left); if (uf.contains(right)) uf.makeUnion(j, right); if (uf.contains(vert)) uf.makeUnion(j, vert); if (uf.size() <= k) result[uf.size() - 1]++; } } for (auto x : result) cout << x << ' '; cout << '\n'; 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 104 105 106 107 108 109 110 111 112 113 114 | #include<cstdio> #include<iostream> #include<iomanip> #include<algorithm> #include<string> #include<vector> #include<cmath> #include<set> #include<unordered_map> using namespace std; typedef vector<int> VI; typedef long long LL; typedef vector<LL> VLL; typedef pair<int, int> PII; typedef vector<PII> VPII; #define FOR(x, b, e) for(int x=b; x<=(e); ++x) #define FORD(x, b, e) for(int x=b; x>=(e); --x) #define REP(x, n) for(int x=0; x<(n); ++x) #define VAR(v, n) __typeof(n) v = (n) #define ALL(c) (c).begin(), (c).end() #define SIZE(x) ((int)(x).size()) #define FOREACH(i, c) for(VAR(i, (c).begin()); i != (c).end(); ++i) #define PB push_back #define ST first #define ND second class UnionFind { private: unordered_map<int, int> id, rank, setSize; int i_size = 0; public: UnionFind() {} ~UnionFind() { id.clear(); rank.clear(); setSize.clear(); } int findSet(int i) { return (id[i] == i) ? i : (id[i] = findSet(id[i])); } bool connected(int i, int j) { return findSet(i) == findSet(j); } int size() { return i_size; } int sizeOfSet(int i) { return setSize[findSet(i)]; } void makeUnion(int i, int j) { if (connected(i, j)) return; int x = findSet(i), y = findSet(j); if (rank[x] > rank[y]) swap(x, y); id[x] = y; if (rank[x] == rank[y]) ++rank[y]; setSize[y] += setSize[x]; --i_size; } bool contains(int i) { return id.find(i) != id.end(); } void add(int i) { rank[i] = 0; id[i] = i; i_size++; } }; int main() { ios_base::sync_with_stdio(0); cin.tie(NULL); cout.tie(NULL); int n, k; cin >> n >> k; vector<VI> canvas(2, VI(n)); VPII positions(2 * n + 1); REP(i, n) { cin >> canvas[0][i]; positions[canvas[0][i]] = { 0, i }; } REP(i, n) { cin >> canvas[1][i]; positions[canvas[1][i]] = { 1, i }; } VI result(k); FOR(i, 1, 2 * n) { result[0]++; UnionFind uf; uf.add(i); FOR(j, i + 1, 2 * n) { uf.add(j); int left = canvas[positions[j].ST][(n + positions[j].ND - 1) % n]; int right = canvas[positions[j].ST][(positions[j].ND + 1) % n]; int vert = canvas[(positions[j].ST + 1) % 2][positions[j].ND]; if (uf.contains(left)) uf.makeUnion(j, left); if (uf.contains(right)) uf.makeUnion(j, right); if (uf.contains(vert)) uf.makeUnion(j, vert); if (uf.size() <= k) result[uf.size() - 1]++; } } for (auto x : result) cout << x << ' '; cout << '\n'; return 0; } |