#include <bits/stdc++.h> using namespace std; typedef unsigned uint; typedef long long ll; typedef unsigned long long ull; typedef long double ldbl; typedef pair<int, int> pii; typedef pair<uint, uint> puu; typedef pair<ll, ll> pll; typedef pair<ull, ull> pull; typedef pair<double, double> pdd; typedef vector<int> vi; typedef vector<uint> vu; typedef vector<ll> vll; typedef vector<ull> vull; typedef vector<pii> vpii; typedef vector<puu> vpuu; typedef vector<pll> vpll; typedef vector<pull> vpull; typedef vector<string> vstr; typedef vector<double> vdbl; typedef vector<ldbl> vldbl; #define pb push_back #define ppb pop_back #define pfr push_front #define ppfr pop_front #define emp emplace #define empb emplace_back #define be begin #define rbe rbegin #define all(x) (x).be(), (x).end() #define rall(x) (x).rbe(), (x).rend() #define fir first #define sec second #define mkp make_pair #define brif(cond) if (cond) break #define ctif(cond) if (cond) continue #define retif(cond) if (cond) return void canhazfast() {ios_base::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);} template<typename T> T gcd(T a, T b) {return b ? gcd(b, a%b) : a;} template<typename T> T extgcd(T a, T b, T &x, T &y) { T x0 = 1, y0 = 0, x1 = 0, y1 = 1; while (b) { T q = a/b; a %= b; swap(a, b); x0 -= q*x1; swap(x0, x1); y0 -= q*y1; swap(y0, y1); } x = x0; y = y0; return a; } int ctz(uint x) {return __builtin_ctz(x);} int ctzll(ull x) {return __builtin_ctzll(x);} int clz(uint x) {return __builtin_clz(x);} int clzll(ull x) {return __builtin_clzll(x);} int popcnt(uint x) {return __builtin_popcount(x);} int popcntll(ull x) {return __builtin_popcountll(x);} int bsr(uint x) {return 31^clz(x);} int bsrll(ull x) {return 63^clzll(x);} #define N 100016 #define K 16 bool vis[2][N]; int ans[N]; int a[2][N]; int row[2*N], col[2*N]; int main() { canhazfast(); int n, k; cin >> n >> k; for (int i : {0, 1}) { for (int j = 0; j < n; ++j) { cin >> a[i][j]; --a[i][j]; row[a[i][j]] = i; col[a[i][j]] = j; } } for (int l = 0; l < 2*n; ++l) { int cnt = 0; for (int i : {0, 1}) fill_n(vis[i], n, false); for (int r = l; r < 2*n; ++r) { int x = row[r], y = col[r]; int yl = y-1, yr = y+1; if (yl < 0) yl = n-1; else if (yr >= n) yr = 0; /// add vis[x][y] = true; ++cnt; if (vis[x^1][y]) --cnt; /// in column if (vis[x][yl]) --cnt; /// to the left if (vis[x][yr]) --cnt; /// to the right if (vis[x^1][y] && vis[x][yl] && vis[x^1][yl]) ++cnt; /// diag left if (vis[x^1][y] && vis[x][yr] && vis[x^1][yr]) ++cnt; /// diag right ++ans[max(cnt, 1)]; } } for (int i = 1; i <= k; ++i) cout << ans[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 | #include <bits/stdc++.h> using namespace std; typedef unsigned uint; typedef long long ll; typedef unsigned long long ull; typedef long double ldbl; typedef pair<int, int> pii; typedef pair<uint, uint> puu; typedef pair<ll, ll> pll; typedef pair<ull, ull> pull; typedef pair<double, double> pdd; typedef vector<int> vi; typedef vector<uint> vu; typedef vector<ll> vll; typedef vector<ull> vull; typedef vector<pii> vpii; typedef vector<puu> vpuu; typedef vector<pll> vpll; typedef vector<pull> vpull; typedef vector<string> vstr; typedef vector<double> vdbl; typedef vector<ldbl> vldbl; #define pb push_back #define ppb pop_back #define pfr push_front #define ppfr pop_front #define emp emplace #define empb emplace_back #define be begin #define rbe rbegin #define all(x) (x).be(), (x).end() #define rall(x) (x).rbe(), (x).rend() #define fir first #define sec second #define mkp make_pair #define brif(cond) if (cond) break #define ctif(cond) if (cond) continue #define retif(cond) if (cond) return void canhazfast() {ios_base::sync_with_stdio(false);cin.tie(nullptr);cout.tie(nullptr);} template<typename T> T gcd(T a, T b) {return b ? gcd(b, a%b) : a;} template<typename T> T extgcd(T a, T b, T &x, T &y) { T x0 = 1, y0 = 0, x1 = 0, y1 = 1; while (b) { T q = a/b; a %= b; swap(a, b); x0 -= q*x1; swap(x0, x1); y0 -= q*y1; swap(y0, y1); } x = x0; y = y0; return a; } int ctz(uint x) {return __builtin_ctz(x);} int ctzll(ull x) {return __builtin_ctzll(x);} int clz(uint x) {return __builtin_clz(x);} int clzll(ull x) {return __builtin_clzll(x);} int popcnt(uint x) {return __builtin_popcount(x);} int popcntll(ull x) {return __builtin_popcountll(x);} int bsr(uint x) {return 31^clz(x);} int bsrll(ull x) {return 63^clzll(x);} #define N 100016 #define K 16 bool vis[2][N]; int ans[N]; int a[2][N]; int row[2*N], col[2*N]; int main() { canhazfast(); int n, k; cin >> n >> k; for (int i : {0, 1}) { for (int j = 0; j < n; ++j) { cin >> a[i][j]; --a[i][j]; row[a[i][j]] = i; col[a[i][j]] = j; } } for (int l = 0; l < 2*n; ++l) { int cnt = 0; for (int i : {0, 1}) fill_n(vis[i], n, false); for (int r = l; r < 2*n; ++r) { int x = row[r], y = col[r]; int yl = y-1, yr = y+1; if (yl < 0) yl = n-1; else if (yr >= n) yr = 0; /// add vis[x][y] = true; ++cnt; if (vis[x^1][y]) --cnt; /// in column if (vis[x][yl]) --cnt; /// to the left if (vis[x][yr]) --cnt; /// to the right if (vis[x^1][y] && vis[x][yl] && vis[x^1][yl]) ++cnt; /// diag left if (vis[x^1][y] && vis[x][yr] && vis[x^1][yr]) ++cnt; /// diag right ++ans[max(cnt, 1)]; } } for (int i = 1; i <= k; ++i) cout << ans[i] << ' '; cout << '\n'; return 0; } |