// source: https://github.com/ShahjalalShohag/code-library/blob/main/Miscellaneous/Schreier%E2%80%93Sims%20algorithm.cpp
#include<bits/stdc++.h>
using namespace std;
using ll=long long;
#define FOR(i,l,r)for(int i=(l);i<=(r);++i)
#define REP(i,n)FOR(i,0,(n)-1)
#define ssize(x)int(x.size())
#ifdef DEBUG
auto&operator<<(auto&o,pair<auto,auto>p){return o<<"("<<p.first<<", "<<p.second<<")";}
auto operator<<(auto&o,auto x)->decltype(x.end(),o){o<<"{";int i=0;for(auto e:x)o<<","+!i++<<e;return o<<"}";}
#define debug(X...)cerr<<"["#X"]: ",[](auto...$){((cerr<<$<<"; "),...)<<endl;}(X)
#else
#define debug(...){}
#endif
#define V(x, y) n*(x) + (y)
const ll mod = 1e9 + 7;
ll fast_exp(ll a, ll x) {
if (x == 0) {
return 1;
}
if (x & 1) {
return fast_exp(a, x-1) * a % mod;
}
ll res = fast_exp(a, x / 2);
return res * res % mod;
}
struct DSU {
int n;
vector<int> rep;
vector<int> ss;
vector<int> up;
vector<int> dwn;
DSU(int _n) {
n = _n;
rep.resize(n*n);
ss.resize(n*n);
up.resize(n*n);
dwn.resize(n*n);
REP(i, n*n) {
rep[i] = i;
ss[i] = 1;
}
}
int find_rep(int a) {
return (a == rep[a]) ? a : rep[a] = find_rep(rep[a]);
}
void union_rep(int a, int b) {
a = find_rep(a);
b = find_rep(b);
if (a == b) {
return;
}
if (ss[a] < ss[b]) {
swap(a, b);
}
rep[b] = a;
ss[a] += ss[b];
}
void calculate() {
REP(i, n) {
REP(j, n) {
int x = find_rep(V(i, j));
if (i < j) {
up[x]++;
}
else {
dwn[x]++;
}
}
}
ll res = 0;
REP(i, n*n) {
// debug(i, up[i], dwn[i]);
res = (res + up[i] * dwn[i] % mod * fast_exp(dwn[i] + up[i] % mod, mod-2)) % mod;
}
cout << res << "\n";
}
};
int main() {
ios_base::sync_with_stdio(0);
int n, k;
cin >> n >> k;
DSU dsu(n);
vector<int> perm(n);
REP(i, k) {
for (int &v : perm) {
cin >> v;
v--;
}
REP(j, n) {
REP(l, n) {
dsu.union_rep(V(j, l), V(perm[j], perm[l]));
}
}
}
dsu.calculate();
}
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 | // source: https://github.com/ShahjalalShohag/code-library/blob/main/Miscellaneous/Schreier%E2%80%93Sims%20algorithm.cpp #include<bits/stdc++.h> using namespace std; using ll=long long; #define FOR(i,l,r)for(int i=(l);i<=(r);++i) #define REP(i,n)FOR(i,0,(n)-1) #define ssize(x)int(x.size()) #ifdef DEBUG auto&operator<<(auto&o,pair<auto,auto>p){return o<<"("<<p.first<<", "<<p.second<<")";} auto operator<<(auto&o,auto x)->decltype(x.end(),o){o<<"{";int i=0;for(auto e:x)o<<","+!i++<<e;return o<<"}";} #define debug(X...)cerr<<"["#X"]: ",[](auto...$){((cerr<<$<<"; "),...)<<endl;}(X) #else #define debug(...){} #endif #define V(x, y) n*(x) + (y) const ll mod = 1e9 + 7; ll fast_exp(ll a, ll x) { if (x == 0) { return 1; } if (x & 1) { return fast_exp(a, x-1) * a % mod; } ll res = fast_exp(a, x / 2); return res * res % mod; } struct DSU { int n; vector<int> rep; vector<int> ss; vector<int> up; vector<int> dwn; DSU(int _n) { n = _n; rep.resize(n*n); ss.resize(n*n); up.resize(n*n); dwn.resize(n*n); REP(i, n*n) { rep[i] = i; ss[i] = 1; } } int find_rep(int a) { return (a == rep[a]) ? a : rep[a] = find_rep(rep[a]); } void union_rep(int a, int b) { a = find_rep(a); b = find_rep(b); if (a == b) { return; } if (ss[a] < ss[b]) { swap(a, b); } rep[b] = a; ss[a] += ss[b]; } void calculate() { REP(i, n) { REP(j, n) { int x = find_rep(V(i, j)); if (i < j) { up[x]++; } else { dwn[x]++; } } } ll res = 0; REP(i, n*n) { // debug(i, up[i], dwn[i]); res = (res + up[i] * dwn[i] % mod * fast_exp(dwn[i] + up[i] % mod, mod-2)) % mod; } cout << res << "\n"; } }; int main() { ios_base::sync_with_stdio(0); int n, k; cin >> n >> k; DSU dsu(n); vector<int> perm(n); REP(i, k) { for (int &v : perm) { cin >> v; v--; } REP(j, n) { REP(l, n) { dsu.union_rep(V(j, l), V(perm[j], perm[l])); } } } dsu.calculate(); } |
English