经过仔细地观察题目,我们可以发现当我们取矩阵中任意一个点时,我们都会得到这一列和这一行的贡献,故能够想到把每次获得的行列加入一个集合,若集合中包含了所有的行列说明我们一定可以构建出这个矩阵。
下面是克鲁斯卡尔的代码
#include "bits/stdc++.h"
//#define int long long
using namespace std;
const int N = 6e3+10;
int fa[N<<1];
struct node{
int x,y,v;
} f[N*N];
int find(int x)
{
if (x == fa[x]) return x;
return fa[x] = find(fa[x]);
}
signed main()
{
ios::sync_with_stdio(false);
int n,m,a,b,c,d,p;
cin >> n >> m >> a >> b >> c >> d >> p;
for (int i = 1; i <= n; ++i) {
for (int j = 1; j <= m; ++j) {
f[(i-1)*m+j] = node{
i,j,(a * a * b + a * c + d)% p};
a = (a * a * b + a * c + d)% p;
}
}
sort(f+1,f+1+n*m,[&](node a1,node a2){
return a1.v < a2.v;
});
for (int i = 1; i <= n+m; ++i) {
fa[i] = i;
}
int res = 0;
for(int i = 1;i <= n*m ;i++)
{
int x = find(f[i].x),y = find(n+f[i].y);
if(x != y)
{
fa[y] = x;
res += f[i].v;
}
}
cout << res << endl;
}