#include<iostream> #include<vector> #include<set> using namespace std; int n, k; vector<pair<int, int> > sasiedzi(int a, int b){ vector<pair<int, int> > w(3); w[1]={a, b^1}; w[0]={a-1, b}; w[2]={a+1, b}; if(a==0){ w[0].first=n-1; } if(a==n-1){ w[2].first=0; } return w; } const int limit=1e3+3; pair<int, int> gdzie[limit];//na jakiej pozycji znajduje sie cos o wartosci i int rep[2*limit]; int ranga[2*limit]; int szukaj(int a){ if(rep[a]==a){ return a; } rep[a]=szukaj(rep[a]); return rep[a]; } void lacz(int a, int b){ int aa=szukaj(a), bb=szukaj(b); if(ranga[aa]>ranga[bb]){ rep[bb]=aa; } if(ranga[aa]<ranga[bb]){ rep[aa]=bb; } if(ranga[aa]==ranga[bb]){ rep[bb]=aa; ranga[aa]++; } } int main(){ cin>>n>>k; int t[n][2]; for(int i=0; i<n; i++) cin>>t[i][0]; for(int i=0; i<n; i++) cin>>t[i][1]; for(int i=0; i<n; i++){ gdzie[t[i][0]]={i, 0}; gdzie[t[i][1]]={i, 1}; } int w[k+1]; fill(w, w+k+1, 0); for(int i=1; i<=2*n; i++){ for(int j=1; j<=2*n; j++){ rep[j]=j; ranga[j]=1; } int spojne=0; for(int j=i; j<=2*n; j++){ //dodaje j-ty i patrze ile roznych spojnych jest auto x=sasiedzi(gdzie[j].first, gdzie[j].second); vector<int> skad;//z jakich spojnych sa dobzi sasiedzi for(auto q:x){ if(t[q.first][q.second]<j && i<=t[q.first][q.second]){//jest dobry skad.push_back(szukaj(t[q.first][q.second])); } } for(auto q:x){ if(t[q.first][q.second]<j && i<=t[q.first][q.second]){//jest dobry lacz(j, t[q.first][q.second]); } } spojne+=1-set<int>(skad.begin(), skad.end()).size(); if(spojne<=k){ w[spojne]++; } //cout<<i<<' '<<j<<'\t'<<spojne<<'\n'; } } for(int i=1; i<=k; i++){ cout<<w[i]<<' '; } }
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 | #include<iostream> #include<vector> #include<set> using namespace std; int n, k; vector<pair<int, int> > sasiedzi(int a, int b){ vector<pair<int, int> > w(3); w[1]={a, b^1}; w[0]={a-1, b}; w[2]={a+1, b}; if(a==0){ w[0].first=n-1; } if(a==n-1){ w[2].first=0; } return w; } const int limit=1e3+3; pair<int, int> gdzie[limit];//na jakiej pozycji znajduje sie cos o wartosci i int rep[2*limit]; int ranga[2*limit]; int szukaj(int a){ if(rep[a]==a){ return a; } rep[a]=szukaj(rep[a]); return rep[a]; } void lacz(int a, int b){ int aa=szukaj(a), bb=szukaj(b); if(ranga[aa]>ranga[bb]){ rep[bb]=aa; } if(ranga[aa]<ranga[bb]){ rep[aa]=bb; } if(ranga[aa]==ranga[bb]){ rep[bb]=aa; ranga[aa]++; } } int main(){ cin>>n>>k; int t[n][2]; for(int i=0; i<n; i++) cin>>t[i][0]; for(int i=0; i<n; i++) cin>>t[i][1]; for(int i=0; i<n; i++){ gdzie[t[i][0]]={i, 0}; gdzie[t[i][1]]={i, 1}; } int w[k+1]; fill(w, w+k+1, 0); for(int i=1; i<=2*n; i++){ for(int j=1; j<=2*n; j++){ rep[j]=j; ranga[j]=1; } int spojne=0; for(int j=i; j<=2*n; j++){ //dodaje j-ty i patrze ile roznych spojnych jest auto x=sasiedzi(gdzie[j].first, gdzie[j].second); vector<int> skad;//z jakich spojnych sa dobzi sasiedzi for(auto q:x){ if(t[q.first][q.second]<j && i<=t[q.first][q.second]){//jest dobry skad.push_back(szukaj(t[q.first][q.second])); } } for(auto q:x){ if(t[q.first][q.second]<j && i<=t[q.first][q.second]){//jest dobry lacz(j, t[q.first][q.second]); } } spojne+=1-set<int>(skad.begin(), skad.end()).size(); if(spojne<=k){ w[spojne]++; } //cout<<i<<' '<<j<<'\t'<<spojne<<'\n'; } } for(int i=1; i<=k; i++){ cout<<w[i]<<' '; } } |