原题传送门:B_Black and white
Prim 算法理解及模板: 传送门
本题大佬详解(思路及Kruskal方法):传送门
本题队友大佬详解及踩坑(Prim方法): 传送门
题目大意:
一个 n ∗ m n*m n∗m 的全白的棋盘,每个格子都有一个权值,涂黑需花费权值。Goodeat 有一个技能,棋盘中任意一个矩形在三个顶角的格子涂黑后,则第四个顶角的格子无需花费权值就能涂黑。问最少花费多少权值全涂黑。
思路:
列举几种方案后,可以发现当每行每列的格子中都有一个涂黑时,在棋盘中就可以构成四个角为三白一黑的矩阵,而这种方案也是涂格子数最少的方案。
于是会想到遍历棋盘的每一行求出最小权值、再遍历每一列求出最小权值,但这样可能会超时。转换思路,由每一行每一列都需要一个格子,可以把每行、每列分别看作一个点,即有 m + n m + n m+n 个点,从这些点中找到最小权值,即最小生成树,连通这些点只需要 m + n − 1 m + n - 1 m+n−1 条边。
这样以来,问题就转换成了最小生成树。在这里,我选择了 P r i m Prim Prim 算法,而没用 K r u s k a l Kruskal Kruskal,是因为用 K r u s k a l Kruskal Kruskal 的时候需要对边进行一次 s o r t sort sort ,对于 2.5 e 7 2.5e7 2.5e7 条边来说效率太低了,但貌似能剪枝?
P r i m Prim Prim算法,时间复杂度( O ² O² 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;
}