AcWing 1144 连接格点

题目描述：

有一个 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

分析：

本题与上一题很类似，图中都是有两类边，处理方法也很类似，考察的其实是想象能力。图中的点是m * n的点阵，连接纵向的两点花费1个单位，连接横向的两点消耗2个单位。我们其实并不需要显式的用一个结构体数组或者邻接矩阵去存储这个图，因为点阵是极具规律性的，自左而右，自上而下点m * n个点依次编号为1到m * n。在读取已有的连线时，获取这两个点的编号，连接这两点即可。后面要考虑的是如何连接剩下的点？

既然连接纵向边的消耗小，我们自然是先枚举纵向边，再枚举横向边，按顺序尝试加入到并查集中。如何枚举纵向边，点阵中的纵向边是点i和点i + n的连线，其中i从1到(m - 1) * n。横向边则是i和i + 1的连线，当遍历到编号为n的倍数的点时跳过即可。具体实现见代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N = 1005;
int n,m,fa[N * N];
int find(int x){
    if(fa[x] != x)  fa[x] = find(fa[x]);
    return fa[x];
}
int main(){
    cin>>m>>n;
    int a,b,x1,y1,x2,y2;
    for(int i = 1;i <= m * n;i++)   fa[i] = i;
    int res = 0;
    while(scanf("%d%d%d%d",&x1,&y1,&x2,&y2) != EOF){
        a = (x1 - 1) * n + y1,b = (x2 - 1) * n +y2;
        fa[find(a)] = find(b);
    }
    int c1 = m * n - n,c2 = c1 + n;
    for(int i = 1;i <= c1;i++){//连接纵向边
        a = find(i),b = find(i + n);
        if(a != b)  fa[a] = b,res += 1;
    }
    for(int i = 1;i <= c2;i++){//连接横向边
        if(i % n == 0)  continue;
        a = find(i),b = find(i + 1);
        if(a != b)  fa[a] = b,res += 2;
    }
    cout<<res<<endl;
    return 0;
}