【模拟试题】连接格点

【模拟试题】连接格点

Description

　　有一个M行N列的点阵，相邻两点可以相连。一条纵向的连线花费一个单位，一条横向的连线花费两个单位。某些点之间已经有连线了，试问至少还需要花费多少个单位才能使所有的点全部连通。

Input

　　第一行输入两个正整数m和n，以下若干行每行四个正整数x1,y1,x2,y2，表示第x1行第y1列的点和第x2行第y2列的点已经有连线。输入保证|x1-x2|+|y1-y2|=1。

Output

　　输出使得连通所有点还需要的最小花费。                                                                                                                                                                                                                                                

Sample Input

2 2
1 1 2 1                                                                                                                                                                                                                                        

Sample Output

3                                                                                                                                                                                                                                                

Hint

【数据规模】：
m,n<=1000                                                                                                                                                                                                                                                

Solution

很明显，在求解MST的过程之中，我们首先连好竖向的边，然后再连横向边。
我们通过并差集解决这个问题。过程分三个阶段，先将已经连好的边加入并差集，然后是枚举所有竖向
相邻的点，最后是横向相连的点。在过程中，如果连边数已经到了n*m-1是，就可以保证存在MST了。
最后输出累加的代价即可。 

#include<iostream>  
#include<cstring>
#include<cstdio>
using namespace std;
inline int read(){
	char c;int rec=0;
	while((c=getchar())<'0'||c>'9');
	while(c>='0'&&c<='9')rec=rec*10+c-'0',c=getchar();
	return rec;
}  
int x1,x2,y1,y2,k,pre[1000002],n,m,ans;  
inline int Pos(int x,int y){return n*(x-1)+y;}  
inline int getfa(int x){  
    if (pre[x]!=x) pre[x]=getfa(pre[x]);  
    return pre[x];  
}  
int main(){   
    m=read();n=read();int x,y;
    for(int i=1;i<=m*n;i++)pre[i]=i;  
    while(scanf("%d%d%d%d",&x1,&y1,&x2,&y2)!=EOF){  
        int p=getfa(Pos(x1,y1)),q=getfa(Pos(x2,y2));  
        if (p!=q){pre[p]=q;k++;}  
    }  
    for(int i=1;i<=n;i++)  
        for(int j=1;j<m;j++){  
        if (k==n*m-1) break;  
        x=getfa(Pos(j,i));y=getfa(Pos(j+1,i));  
        if (x!=y){pre[x]=y;k++;ans++;}  
    }  
    for(int j=1;j<=m;j++)  
        for(int i=1;i<n;i++){  
        if (k==n*m-1) break;  
        x=getfa(Pos(j,i));y=getfa(Pos(j,i+1));  
        if (x!=y){pre[x]=y;k++;ans+=2;}  
    }cout<<ans;
	return 0;  
}