连接格点
题目描述
核心思路
通过阅读题目,设有 n n n行 m m m列,那么总共有 n × m n\times m n×m个点,就相当于图中有 n × m n\times m n×m个顶点。然后想用一些边,来使得所有顶点都能够连通,并且这些边权之和最小。那么这与最小生成树的含义是相同的。因此,透过题意我们知道其实就是想要求最小生成树。
但是题目还是有点特殊的,一条纵向的连线花费一个单位,一条横向的连线花费两个单位,也就是说边权只能是1或者2,而且某些点之间已经有连线了,也就是说有些顶点已经有边连接起来了。
我们将已有连线的这些点所形成的各个连通块都分别看作是一个个"新的节点",设有x个"新的节点",现在需要用 边 来连接这x个"新的节点",并且使得所用边的权值总和最小。有x个"新的节点",那么就需要x-1条边才能来连通,因为竖边的花费少,所以就先尽量用竖边,然后剩下的用横边,也就是贪心。
由于有连通块和并查集,因此我们使用Kruskal算法来求解最小生成树。但是我们这里并不对所有边排序。因为:有 n × ( m − 1 ) n\times(m-1) n×(m−1)条横边,有 m × ( n − 1 ) m\times(n-1) m×(n−1)条竖边,因此最多有 n × ( m − 1 ) + m × ( n − 1 ) = 2 n m − n − m n\times(m-1)+m\times(n-1)=2nm-n-m n×(m−1)+m×(n−1)=2nm−n−m, n n n和 m m m最坏取到 1000 1000 1000,所以最多有 2 × 1000 × 1000 − 1000 − 1000 = 1998000 2\times1000\times1000-1000-1000=1998000 2×1000×1000−1000−1000=1998000,约两百万条边。如果给这两百万条边排序的话,时间其实也是可以的,但是这里提供一种更加优化的方法:
由于Kruskal算法是排序后,优先选择边权小的那些边,然后再选择边权大的那些边。那么我们这里这么做:
优先挑选了边权为1的边作为最小生成树中的边,然后再挑选边权为2的边作为最小生成树中的边,可以发现这种做法中选边也是按边权从小到大进行,这其实就等效于排序了。
因此,我们只需要先尽量选择那些边权为1的竖边来建图,然后再选择那些边权为2的横边来建图,那么就可以不需要排序了,这样就省去了给所有边排序的时间。
代码
/*
* 在求解MST的过程之中,我们首先连好竖向的边,然后再连横向边.我们通过并差集解决这个问题.
* 我们将已有连线的这些点所形成的各个连通块都分别看作是一个个"新的节点",设有x个"新的节点"
* 现在需要用 边 来连接这x个"新的节点",并且使得所用边的权值总和最小
* 有x个"新的节点",那么就需要x-1条边才能来连通,因为竖边的花费少,所以就先尽量用竖边,然后剩下的用横边,也就是贪心
* 过程分三个阶段:
* (1)先将已经连好的边加入并差集
* (2)然后是枚举所有竖向相邻的点
* (3)最后是横向相连的点
*/
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
//N是点的数量 M的边的数量
const int A=1010,N=A*A,M=2*A*A;
int n,m;
int p[N];
//将二维坐标转换成一维编号
inline int get(int x,int y)
{
return (x-1)*m+y;
}
int find(int x)
{
if(x!=p[x])
p[x]=find(p[x]);
return p[x];
}
int main()
{
int res=0;
scanf("%d%d",&n,&m);
for(int i=1;i<=n*m;i++)
p[i]=i;
int x1,y1,x2,y2;
while(~scanf("%d%d%d%d",&x1,&y1,&x2,&y2))
{
int a=get(x1,y1);
int b=get(x2,y2);
p[find(a)]=find(b);
}
//这里进行Kruskal算法时 为什么不需要对边排序呢?
//因为我们是优先挑选了边权为1的边作为最小生成树中的边,然后再挑选边权为2的边作为最小生成树中的边
//因此其实就相当于对边权排序了,因为Kruskal算法也是按边权从小到大先选择边权小的边,然后再选择边权大的边
//先处理边权为1的竖边
//这里不能写成i<=n 因为我们用到了i+1 因此当i=n时,i+1就是n+1了,就会越界
for(int i=1;i<n;i++)
{
for(int j=1;j<=m;j++)
{
//a和b这上下两个点就构成了一条竖边
int a=get(i,j);
int b=get(i+1,j);
int pa=find(a);
int pb=find(b);
if(pa!=pb)
{
//累加边权为1的竖边
res+=1;
p[pa]=pb;
}
}
}
//再处理边权为2的横边
//这里不能写成j<=m 因为我们用到了j+1 因此当j=m时,j+1就是m+1了,就会越界
for(int i=1;i<=n;i++)
{
for(int j=1;j<m;j++)
{
//a和b这左右两个点就构成了一条横边
int a=get(i,j);
int b=get(i,j+1);
int pa=find(a);
int pb=find(b);
if(pa!=pb)
{
//累加边权为2的横边
res+=2;
p[pa]=pb;
}
}
}
printf("%d\n",res);
return 0;
}