题意:一个矩阵中,一个点与其上下左右均有边,边权为差值,求最小生成树。
思路:建好图,套克鲁斯卡尔。
数组大小开错,找了一个小时。跟个智障一样。
#include<bits\stdc++.h>
using namespace std;
const int maxn = 2000;
int mp[maxn][maxn];
int f[1000*1000+10]; //错误点
struct edge
{
int from,to,w;
};
vector<edge> e;
void addedge(int i,int j,int w)
{
e.push_back(edge{i,j,w});
}
int Find(int x)
{
return f[x]==x?x:f[x]=Find(f[x]);
}
bool cmp(edge a,edge b)
{
return a.w<b.w;
}
int kruscal(int n)
{
int ans=0,num=0;
for(int i=0;i<n;i++)
f[i]=i;
for(int i=0;i<e.size();i++)
{
edge temp=e[i];
int u=temp.from;int v=temp.to;
int xx=Find(u);
int yy=Find(v);
if(xx!=yy)
{
f[xx]=yy;
ans+=e[i].w;
num++;
if(num==n-1)break;
}
}
return ans;
}
int main()
{
int t;
// freopen("in.txt","r",stdin);
// freopen("out1.txt","w",stdout);
scanf("%d",&t);
for(int cas=1;cas<=t;cas++)
{
int n,m;
e.clear();
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
scanf("%d",&mp[i][j]);
for(int i=0;i<n;i++)
{
for(int j=0;j<m;j++)
{
if(j!=m-1)
addedge(i*m+j,i*m+j+1,abs(mp[i][j]-mp[i][j+1]));
if(i!=n-1)
addedge(i*m+j,(i+1)*m+j,abs(mp[i][j]-mp[i+1][j]));
}
}
sort(e.begin(),e.end(),cmp);
int ans=kruscal(n*m);
printf("Case #%d:\n%d\n",cas,ans);
}
}