B题题解
队友别骂了,别骂了,我不该以为这题是最短路
思路
为了实现最少花费,需满足:在我们涂完若干个点后,其他的点对总花费不再有贡献(涂黑这些点时不花钱了)
我们会发现 :最少,我们需要涂黑 n + m - 1 个点,且涂完这些点后 所有的行 和所有的列 会都在一个联通块里面
举例 如下:
不妨令 n = 2,m = 2,此时我们最少需要涂 3个点
令 g[1][1] = g[1][2] = g[2][1] = 1,g[2][2] = 3
为了最少花费,我们要涂的三个点显然是:(1,1),(1,2),(2,1)
涂完(1,1)后,我们把 行1 和 列1 放到一个联通块里面
涂完(1,2)后,我们把 行1 和 列2 放到一个联通块里面
涂完(2,1)后,我们把 行2 和 列1 放到一个联通块里面
到此,所有的行 和 所有的列 都在一个联通块里了,同时我们也实现了最少花费
在这个例子中,涂黑任意三个点都可以实现 把所有的行 和 所有的列 都放在一个联通块里
但为了 最少花费,所以我们选择的是 (1,1),(1,2),(2,1) 这三个点
我们提炼出两个关键字 联通块、最小花费:连通块(并查集)+ 最小花费 = Kruskal(最小生成树)
所以该题使用 Kruskal 求最小(花费)生成树
Code如下
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 3e7+10;
const int M = 5010;
struct node{
int x,y;
ll v;
bool operator < (const node & u) const { return v < u.v; }
}A[N];
int fa[2*M];
int n,m,a,b,c,d,p;
void Init(){
A[0].v=a;
for(int i=1;i<=n;i++) for(int j=1;j<=m;j++) {
int id=m*(i-1)+j;
A[id]={i,j,((ll)A[id-1].v*A[id-1].v*b+(ll)A[id-1].v*c+d)%p};
}
}
int find(int x){
if(x==fa[x]) return x;
return fa[x]=find(fa[x]);
}
void kruskal(){
//选若干个点(最少n+m-1个),使得所有行,所有列在一个联通块里面
ll res=0,cnt=n+m-1;
for(int i=1;i<=n+m;i++) fa[i]=i; // 1~n 表示行,n+1~n+m 表示列
sort(A+1,A+1+m*n);
for(int i=1;i<=m*n;i++){
int x=A[i].x,y=A[i].y,v=A[i].v;
x=find(x),y=find(n+y);
if(!cnt) break;
// 按理说 没有 cnt 这个变量 应该也可以,但是去掉后 只能过90% 剩下的10% 会 T,不知道为啥
if(x!=y){
fa[x]=y;
res+=v;
cnt--;
}
}
cout<<res<<endl;
}
int main(){
cin>>n>>m>>a>>b>>c>>d>>p;
Init();
kruskal();
return 0;
}
京公网安备 11010502036488号