#include <bits/stdc++.h> using namespace std; using ll = long long; using db = double; using ull = unsigned long long; using pi = pair<int, int>; using pl = pair<ll, ll>; using vi = vector<int>; using vl = vector<ll>; using vpi = vector<pi>; using vpl = vector<pl>; #define pb push_back #define eb emplace_back #define mp make_pair #define f first #define s second #define tcT template<class T #define tcTU tcT, class U #define sz(x) int((x).size()) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define rep(i,a,b) for(int i = (a); i < (b); i++) #define rrep(i,a,b) for(int i = (b) - 1; i >= (a); i--) const db PI = acos(-1); mt19937 rng((uint32_t)chrono::steady_clock::now().time_since_epoch().count()); tcT> bool ckmin(T& a, const T& b) { return b < a ? a=b, 1 : 0; } tcT> bool ckmax(T& a, const T& b) { return a < b ? a=b, 1 : 0; } tcT> void _dbg(const char *sdbg, T h){ cerr<<sdbg<<'='<<h<<endl; } tcT, class... TA> void _dbg(const char *sdbg, T h, TA... a) { while(*sdbg!=',') cerr<<*sdbg++; cerr<<'='<<h<<','; _dbg(sdbg+1, a...); } tcT> ostream &operator<<(ostream& os, vector<T> V) { os << "["; for (auto vv : V) os << vv << ","; return os << "]"; } tcT, class V> ostream &operator<<(ostream &os, pair<T,V> P) { return os << "(" << P.f << "," << P.s << ")"; } #ifdef LOCAL #define debug(...) _dbg(#__VA_ARGS__, __VA_ARGS__) #else #define debug(...) (__VA_ARGS__) #define cerr if(0)cout #endif const int MAXN = 1e5+10; int a[MAXN][2]; pi place[2*MAXN]; bool vis[2*MAXN]; const int MAXK = 12; int res[MAXK]; constexpr pi moves[] = {mp(-1,0), mp(1,0), mp(0,1)}; int n, k; inline pi operator+(const pi &a, const pi &b) { return mp((a.f + b.f + n) % n, (a.s+b.s) % 2); } int dsu[2*MAXN]; int siz[2*MAXN]; int connected_components; void reset_dsu() { iota(dsu, dsu+2*MAXN, 0); fill(siz, siz+2*MAXN, 1); connected_components = 0; } int fnd(int x) { return dsu[x] == x ? x : dsu[x] = fnd(dsu[x]); } void uni(int a, int b) { a = fnd(a); b = fnd(b); if (a == b) return; if (siz[a] > siz[b]) swap(a, b); siz[b] += siz[a]; dsu[a] = b; connected_components--; } signed main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cin >> n >> k; rep(j,0,2) { rep(i,0,n) { cin >> a[i][j]; a[i][j]--; place[a[i][j]] = mp(i,j); } } rep(l,0,2*n) { reset_dsu(); rep(r,l,2*n) { connected_components++; pi cur = place[r]; for (pi mv : moves) { pi nxt = cur + mv; int v = a[nxt.f][nxt.s]; if (l <= v && v <= r) { uni(v, r); } } if (connected_components <= k) { res[connected_components]++; } } } rep(i,1,k+1) { cout << res[i] << " "; } 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 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 | #include <bits/stdc++.h> using namespace std; using ll = long long; using db = double; using ull = unsigned long long; using pi = pair<int, int>; using pl = pair<ll, ll>; using vi = vector<int>; using vl = vector<ll>; using vpi = vector<pi>; using vpl = vector<pl>; #define pb push_back #define eb emplace_back #define mp make_pair #define f first #define s second #define tcT template<class T #define tcTU tcT, class U #define sz(x) int((x).size()) #define all(x) (x).begin(), (x).end() #define rall(x) (x).rbegin(), (x).rend() #define rep(i,a,b) for(int i = (a); i < (b); i++) #define rrep(i,a,b) for(int i = (b) - 1; i >= (a); i--) const db PI = acos(-1); mt19937 rng((uint32_t)chrono::steady_clock::now().time_since_epoch().count()); tcT> bool ckmin(T& a, const T& b) { return b < a ? a=b, 1 : 0; } tcT> bool ckmax(T& a, const T& b) { return a < b ? a=b, 1 : 0; } tcT> void _dbg(const char *sdbg, T h){ cerr<<sdbg<<'='<<h<<endl; } tcT, class... TA> void _dbg(const char *sdbg, T h, TA... a) { while(*sdbg!=',') cerr<<*sdbg++; cerr<<'='<<h<<','; _dbg(sdbg+1, a...); } tcT> ostream &operator<<(ostream& os, vector<T> V) { os << "["; for (auto vv : V) os << vv << ","; return os << "]"; } tcT, class V> ostream &operator<<(ostream &os, pair<T,V> P) { return os << "(" << P.f << "," << P.s << ")"; } #ifdef LOCAL #define debug(...) _dbg(#__VA_ARGS__, __VA_ARGS__) #else #define debug(...) (__VA_ARGS__) #define cerr if(0)cout #endif const int MAXN = 1e5+10; int a[MAXN][2]; pi place[2*MAXN]; bool vis[2*MAXN]; const int MAXK = 12; int res[MAXK]; constexpr pi moves[] = {mp(-1,0), mp(1,0), mp(0,1)}; int n, k; inline pi operator+(const pi &a, const pi &b) { return mp((a.f + b.f + n) % n, (a.s+b.s) % 2); } int dsu[2*MAXN]; int siz[2*MAXN]; int connected_components; void reset_dsu() { iota(dsu, dsu+2*MAXN, 0); fill(siz, siz+2*MAXN, 1); connected_components = 0; } int fnd(int x) { return dsu[x] == x ? x : dsu[x] = fnd(dsu[x]); } void uni(int a, int b) { a = fnd(a); b = fnd(b); if (a == b) return; if (siz[a] > siz[b]) swap(a, b); siz[b] += siz[a]; dsu[a] = b; connected_components--; } signed main() { ios_base::sync_with_stdio(false); cin.tie(nullptr); cin >> n >> k; rep(j,0,2) { rep(i,0,n) { cin >> a[i][j]; a[i][j]--; place[a[i][j]] = mp(i,j); } } rep(l,0,2*n) { reset_dsu(); rep(r,l,2*n) { connected_components++; pi cur = place[r]; for (pi mv : moves) { pi nxt = cur + mv; int v = a[nxt.f][nxt.s]; if (l <= v && v <= r) { uni(v, r); } } if (connected_components <= k) { res[connected_components]++; } } } rep(i,1,k+1) { cout << res[i] << " "; } cout << "\n"; return 0; } |