#include <cstdio>
#include <vector>
#include <algorithm>
#include <string>
#include <cstring>
#include <sstream>
#include <iostream>
#include <map>
#include <cmath>
#include <set>
#include <queue>
#include <numeric>
#include <ctime>
using namespace std;
typedef long long LL;
#define INF 1000000000
#define SZ(v) ((int)(v).size())
#define FOR(i,a,b) for(int i=(a);i<(b);i++)
#define FORE(i,a,b) for(int i=(a);i<=(b);i++)
#define FS first
#define SD second
#define MP make_pair
#define ZERO(t) memset(t, 0, sizeof(t))
#define MINUS(t) memset(t, -1, sizeof(t))
#define MOD 1000000007
#define MIN(_a,_b) ((_a)<(_b)?(_a):(_b))
#define MAX(_a,_b) ((_a)>(_b)?(_a):(_b))
int UN[200005];
int N[200005][3];
int T[200005][2];
int find(int x){
return UN[x]==x?x:UN[x]=find(UN[x]);
}
void join(int x,int y){
UN[find(x)]=find(y);
}
LL result[12];
int CHECK[4];
int check(int n) {
int res = n;
FOR(i,0,n) {
FOR(j,i+1,n) {
if (CHECK[i] == CHECK[j]) {
--res;
break;
}
}
}
return res;
}
int main() {
int n, k; scanf("%d%d", &n, &k);
FOR(i,0,2) {
FOR(j,0,n) {
scanf("%d", &T[j][i]);
}
}
FOR(i,0,2) {
FOR(j,0,n) {
int v = T[j][i];
N[v][0] = T[j][(i + 1) % 2];
N[v][1] = T[(j + 1) % n][i];
N[v][2] = T[(j - 1 + n) % n][i];
}
}
FORE(a,1,2*n) {
int cur = 0;
FORE(b,a,2*n) UN[b] = b;
FORE(b,a,2*n) {
int cc = 0;
FOR(i,0,3) {
int nei = N[b][i];
if (nei >= a && nei <= b) CHECK[cc++] = find(nei);
}
int before = check(cc);
cc = 0;
FOR(i,0,3) {
int nei = N[b][i];
if (nei >= a && nei <= b) join(nei, b);
}
FOR(i,0,3) {
int nei = N[b][i];
if (nei >= a && nei <= b) CHECK[cc++] = find(nei);
}
CHECK[cc++] = find(b);
int after = check(cc);
cur += (after - before);
if (cur <= k) result[cur]++;
}
}
FORE(i,1,k) printf("%lld ", result[i]);
printf("\n");
}
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 <vector> #include <algorithm> #include <string> #include <cstring> #include <sstream> #include <iostream> #include <map> #include <cmath> #include <set> #include <queue> #include <numeric> #include <ctime> using namespace std; typedef long long LL; #define INF 1000000000 #define SZ(v) ((int)(v).size()) #define FOR(i,a,b) for(int i=(a);i<(b);i++) #define FORE(i,a,b) for(int i=(a);i<=(b);i++) #define FS first #define SD second #define MP make_pair #define ZERO(t) memset(t, 0, sizeof(t)) #define MINUS(t) memset(t, -1, sizeof(t)) #define MOD 1000000007 #define MIN(_a,_b) ((_a)<(_b)?(_a):(_b)) #define MAX(_a,_b) ((_a)>(_b)?(_a):(_b)) int UN[200005]; int N[200005][3]; int T[200005][2]; int find(int x){ return UN[x]==x?x:UN[x]=find(UN[x]); } void join(int x,int y){ UN[find(x)]=find(y); } LL result[12]; int CHECK[4]; int check(int n) { int res = n; FOR(i,0,n) { FOR(j,i+1,n) { if (CHECK[i] == CHECK[j]) { --res; break; } } } return res; } int main() { int n, k; scanf("%d%d", &n, &k); FOR(i,0,2) { FOR(j,0,n) { scanf("%d", &T[j][i]); } } FOR(i,0,2) { FOR(j,0,n) { int v = T[j][i]; N[v][0] = T[j][(i + 1) % 2]; N[v][1] = T[(j + 1) % n][i]; N[v][2] = T[(j - 1 + n) % n][i]; } } FORE(a,1,2*n) { int cur = 0; FORE(b,a,2*n) UN[b] = b; FORE(b,a,2*n) { int cc = 0; FOR(i,0,3) { int nei = N[b][i]; if (nei >= a && nei <= b) CHECK[cc++] = find(nei); } int before = check(cc); cc = 0; FOR(i,0,3) { int nei = N[b][i]; if (nei >= a && nei <= b) join(nei, b); } FOR(i,0,3) { int nei = N[b][i]; if (nei >= a && nei <= b) CHECK[cc++] = find(nei); } CHECK[cc++] = find(b); int after = check(cc); cur += (after - before); if (cur <= k) result[cur]++; } } FORE(i,1,k) printf("%lld ", result[i]); printf("\n"); } |
English