2021牛客暑期多校训练营3 black and white（最小生成树）

题目大意
古德特有一个有n行m列的白色棋盘。
每个格子（i，j）有一个权重c（i，j）。在任何时候，网格（i，j）都可以被染成黑色，代价是c（i，j）。
古德特有一个特殊的才能。对于任意两行两列的四个相交方格，如果其中三个是黑色方格，古德特可以不计成本地将第四个方格染成黑色。
请找出染黑棋盘的最低成本。
由于格子的数量很多，我们用以下方法来产生权重。
思路：
首先我们来证明需要遍历的点数

我们可以发现，当染色点所在位置经过平移可以在图中心形成一个十字时，就可以将整个图全部染色，那么也就是说，我们至少需要n+m-1个点，那么我们就对这满足要求的点来进行最小生成树操作即可。

但是如果贸然的进行操作，会使得某点可能会重复使用，我们要用二分图的方法来存图，将行号和列号分开，并且连边即可，

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<map>
#include<cstdio>
#include<cmath>
#include<iomanip>
#include<sstream>
#include<queue>

#define pi 3.1415926535
using namespace std;
typedef long long ll;
const int mod = 1e9 + 7;
typedef pair<int, int> pii;
const int N = 1e6 + 10;

map<int, int>mp;
int s[5010*2][5010*2];
int dis[N];
bool st[N];
int n, m, a, b, c, d, p;

int prim() {
	memset(dis, 0x3f, sizeof dis);
	dis[1] = 0;
	int res = 0;
	for (int i = 0; i < n+m; i++) {
		int t = -1;
		for (int j = 1; j <= n+m; j++)
			if (!st[j] && (t == -1 || dis[t] > dis[j]))
				t = j; 

		if (i)res += dis[t];
		st[t] = true;
		for (int j = 1; j <= n+m; j++)dis[j] = min(dis[j], s[t][j]);
	}
	return res;
}

int main() {
	cin >> n >> m >> a >> b >> c >> d >> p;
	memset(s, 0x3f, sizeof s);
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++) {
			a = ((ll)a * a * b + (ll)a * c + d) % p;
			s[i][j + n] = s[j + n][i] = a;
		}
	}
	/*for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= m; j++)
		{
			cout << s[i][j] << ' ';
		}
		cout << endl;
	}*/
	cout << prim() << endl;
}