牛客多校day3-B.Black and white

Black and white

题意：

给你一个矩阵，原本都是白色。你可以将一个格子改成黑色，花费$c_{i,j}$。如果两行两列相交的$4$个格子有$3$个染成黑色了，就能免费把第四个染成黑色。问最少要多少代价才能将整个棋盘变黑。

思路：

如果我们把$(i,j)$相连，就会发现。如果两行两列$4$个格子有$3$个染黑的时候，就形成了一颗树的样子。
所以就是将行列拆开然后求最小生成树。
因为是一个完全图，所以用prim算法。

代码：

#include<bits/stdc++.h>
#define fi first
#define se second
#define ll long long
#define mp make_pair
#define pb push_back
#define ls x<<1
#define rs x<<1|1
#define lson x<<1,l,mid
#define rson x<<1|1,mid+1,r
#define pii pair<int,int>
#define all(x) x.begin(),x.end()
#define cl(x,y) memset(x,y,sizeof(x))
#define nxtp(a,n) next_permutation(a+1,a+n+1)
#define mem(x,y,n) memset(x,y,sizeof(int)*(n+5))
const int N=1e4+10;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const double eps=1e-8;
const double pi=acos(-1);
const double INF=1e18;
using namespace std;

int di[N][N],dis[N],vis[N];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,m,a,b,c,d,p;
	cin>>n>>m>>a>>b>>c>>d>>p;
	cl(di,inf);
	cl(dis,inf);
	int i,j,sum=a;
	for(i=1;i<=n;i++)
		for(j=1;j<=m;j++)
		{
			ll w=((ll)sum*(ll)sum%p*(ll)b%p+(ll)sum*(ll)c%p+(ll)d)%p;
			di[i][j+n]=di[j+n][i]=w;
			sum=w;
		}
	int ans=0,mi,tot=0,k;
	for(i=1;i<=n+m;i++)
		dis[i]=di[1][i];
	vis[1]=1;
	while(tot<n+m-1)
	{
		mi=inf;
		tot++;
		for(i=1;i<=n+m;i++)
			if(!vis[i] && dis[i]<mi)
			{
				mi=dis[i];
				k=i;
			}
		ans+=mi;
		for(i=1;i<=n+m;i++)
			if(dis[i]>di[k][i] && !vis[i])
				dis[i]=di[k][i];
		vis[k]=1;
	}
	cout<<ans<<endl;
	return 0;
}