2021牛客暑期多校训练营3 B_Black and white

原题传送门：B_Black and white

---

Prim 算法理解及模板： 传送门
本题大佬详解（思路及Kruskal方法）：传送门
本题队友大佬详解及踩坑（Prim方法）： 传送门

---

题目大意：

一个 $n\ast m$ 的全白的棋盘，每个格子都有一个权值，涂黑需花费权值。Goodeat 有一个技能，棋盘中任意一个矩形在三个顶角的格子涂黑后，则第四个顶角的格子无需花费权值就能涂黑。问最少花费多少权值全涂黑。

思路：

列举几种方案后，可以发现当每行每列的格子中都有一个涂黑时，在棋盘中就可以构成四个角为三白一黑的矩阵，而这种方案也是涂格子数最少的方案。

于是会想到遍历棋盘的每一行求出最小权值、再遍历每一列求出最小权值，但这样可能会超时。转换思路，由每一行每一列都需要一个格子，可以把每行、每列分别看作一个点，即有 $m+n$ 个点，从这些点中找到最小权值，即最小生成树，连通这些点只需要 $m+n-1$ 条边。

这样以来，问题就转换成了最小生成树。在这里，我选择了 $Prim$ 算法，而没用$Kruskal$，是因为用 $Kruskal$ 的时候需要对边进行一次 $sort$ ，对于 $2.5e7$ 条边来说效率太低了，但貌似能剪枝？

$Prim$算法，时间复杂度（$O²$)

AC代码：

#include<bits/stdc++.h>
using  namespace std;
#define MAX 10001
#define MAXCOST 0x7fffffff
using ll = long long;
int graph[MAX][MAX];
bool a[MAX];

ll prim(int graph[][MAX], int n)
{
	int lowcost[MAX];
	int mst[MAX];
	memset(a, false, sizeof a);
	ll i, j, min, minid, sum = 0;
	for (i = 2; i <= n; i++)
	{
		lowcost[i] = graph[1][i];
		mst[i] = 1;
	}
	mst[1] = 0;
	a[1] = true;

	for (i = 2; i <= n; i++)
	{
		min = MAXCOST;
		minid = 0;
		for (j = 2; j <= n; j++)
		{
			if (lowcost[j] < min && a[j] == false)
			{
				min = lowcost[j];
				minid = j;
			}
		}
		//cout << "V" << mst[minid] << "-V" << minid << "=" << min << endl;
		sum += min;
		a[minid] = true;
		for (j = 2; j <= n; j++)
		{
			if (graph[minid][j] < lowcost[j])
			{
				lowcost[j] = graph[minid][j];
				mst[j] = minid;
			}
		}
	}
	return sum;
}

int main()
{
	int n, m, a, b, c, d, p;
	cin >> n >> m >> a >> b >> c >> d >> p;
	if (n == 1 && m == 1) {
		cout << ((ll)a * a * b + (ll)a * c + d) % p;
		return 0;
	}
	for (int i = 1; i < MAX; i++)
	{
		for (int j = 1; j < MAX; j++)
		{
			graph[i][j] = MAXCOST;
		}
	}
	for (int i = 1; i <= n; i++)
		for (int j = 1; j <= m; j++) {
			a = ((ll)a * a * b + (ll)a * c + d) % p;
			graph[i + m][j] = graph[j][i + m] = a;
		}
	cout << prim(graph, n + m);
	return 0;
}