连接格点(最小生成树)
时间: 1000ms / 空间: 65536KiB / Java类名: Main
描述
Problem 2 : grid
连接格点
问题描述:
有一个M行N列的点阵,相邻两点可以相连。一条纵向的连线花费一个单位,一条横向的连线花费两个单位。某些点之间已经有连线了,试问至少还需要花费多少个单位才能使所有的点全部连通。
输入格式
第一行输入两个正整数m和n。
以下若干行每行四个正整数x1,y1,x2,y2,表示第x1行第y1列的点和第x2行第y2列的点已经有连线。输入保证|x1-x2|+|y1-y2|=1。
输出格式
输出使得连通所有点还需要的最小花费。
测试样例1
输入
输入样例
2 2
1 1 2 1
输出
输出样例
3
写在前面:最小生成树所花时间最长的题目
——————————————————————————————————————————————
解题思路:刚开始想法就是先连行再连列,但我脑(fan)洞(sha)大(bi)开,写出了个二维冰茶几,写的乱七八糟,只过了30%的数据,后来问了问Shallwe大爷,知道应该转换成一维(其实我也这么想过,但是没敢去试(╯°口°)╯︵┴─┴),再进行并查集操作就可以了),然后就过了。(其实我之前想用数学方法做,但是写完以后发现不对……)
代码:
#include<cstdio>
#include<iostream>
using namespace std;
int x1,x2,y1,y2,k,father[1000002],n,m,ans;
int f(int x,int y)
{
return n*(x-1)+y;
}
int find(int x)
{
if (father[x]!=x) father[x]=find(father[x]);
return father[x];
}
main()
{
scanf("%d%d",&m,&n);
for (int i=1;i<=m*n;i++)
father[i]=i;
while (cin>>x1>>y1>>x2>>y2)
{
int p=find(f(x1,y1)),q=find(f(x2,y2));
if (p!=q) {father[p]=q;k++;}
}
for (int i=1;i<=n;i++)
for (int j=1;j<m;j++)
{
if (k==n*m-1) break;
int x,y;
x=find(f(j,i));y=find(f(j+1,i));
if (x!=y){father[x]=y;k++;ans++;}
}
for (int j=1;j<=m;j++)
for (int i=1;i<n;i++)
{
if (k==n*m-1) break;
int x,y;
x=find(f(j,i));y=find(f(j,i+1));
if (x!=y){father[x]=y;k++;ans+=2;}
}
printf("%d",ans);
}