hdu 5253 连接的管道 最小生成树

题意：一个矩阵中，一个点与其上下左右均有边，边权为差值，求最小生成树。

思路：建好图，套克鲁斯卡尔。

数组大小开错，找了一个小时。跟个智障一样。

#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);
	}
}