有一个 m 行 n 列的点阵,相邻两点可以相连。
一条纵向的连线花费一个单位,一条横向的连线花费两个单位。
某些点之间已经有连线了,试问至少还需要花费多少个单位才能使所有的点全部连通。
输入格式
第一行输入两个正整数 m 和 n。
以下若干行每行四个正整数 x1,y1,x2,y2,表示第 x1 行第 y1 列的点和第 x2 行第 y2 列的点已经有连线。
输入保证|x1−x2|+|y1−y2|=1。
输出格式
输出使得连通所有点还需要的最小花费。
数据范围
1≤m,n≤1000
0≤已经存在的连线数≤10000
输入样例:
2 2
1 1 2 1
输出样例:
3
解析:
先把二维的坐标转化成一维上。
把所有关系的两点连起来。
处理每个点,要优先处理权重小的。因为这样可以免去排序的时间复杂度。
最后枚举这些点,即可。
#include<bits/stdc++.h>
using namespace std;
const int N=1000000;
int m,n;
int x1,y11,x2,y2;
struct node
{
int u,v,w;
}e[N*4];
int fa[N*4];
int ids[1005][1005];
int t;
int dx[4] = {-1, 0, 1, 0}, dy[4] = {0, 1, 0, -1}, dw[4] = {1, 2, 1, 2};
int cnt;
void get_edge()
{
for(int u=0;u<2;u++) //遍历纵向 和横向 。因为纵向的权重比横向的权重小
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
for(int k=0;k<4;k++) //遍历四个方向
{
if(k%2==u)
{
int x=i+dx[k];
int y=j+dy[k];
int w=dw[k];
if(x>=1&&x<=m&&y>=1&&y<=n)
{
int a=ids[i][j];
int b=ids[x][y];
if(a<b)
e[++cnt]={a,b,w};
}
}
}
}
int find(int x)
{
if(fa[x]!=x) return fa[x]=find(fa[x]);
return fa[x];
}
int main()
{
cin>>m>>n;
for(int i=1;i<=m;i++)
for(int j=1;j<=n;j++)
{
ids[i][j]=++t;
}
for(int i=1;i<=n*m;i++) fa[i]=i;
while(cin>>x1>>y11>>x2>>y2)
{
int a=ids[x1][y11];
int b=ids[x2][y2];
int x=find(a);
int y=find(b);
if(x!=y)
{
fa[x]=y;
}
}
get_edge(); //处理每个点
int res=0;
for(int i=1;i<=cnt;i++) //枚举点
{
int a=e[i].u;
int b=e[i].v;
int c=e[i].w;
int x=find(a);
int y=find(b);
if(x!=y)
{
fa[x]=y;
res+=c;
}
}
cout<<res<<endl;
}