题目:
农夫FJ的奶牛们有空旷恐惧症,所以,FJ打算在他的农场围上篱笆。他的农场是一个矩形区域。左上角的坐标是(0,0),右下角的坐标是(A,B),FJ修建了n(0<=n<=2000)个竖直的篱笆,其横坐标分别为a1,a2,a3,……,an,其中0<ai<A,,每一个篱笆从(ai,0)到(ai,B)也修建了m个水平的篱笆,其纵坐标为b1,b2,b3,……,bm,其中0<bi<B,每一个篱笆从(0,bi)到(A,bi)。这些篱笆把整个农场分成(n+1)*(m+1)个区域。
不幸的是FJ忘了在篱笆上装门了。这导致奶牛无法在不同的区域之间移动。于是他决定将某些篱笆拆掉。现在要使得所有的区域都联通,请问最少要拆掉多长的篱笆。
比如下面这个篱笆
+---+--+
| | |
+---+--+
| | |
| | |
+---+--+
可以这么拆:
+---+--+
| |
+---+ +
| |
| |
+---+--+
乍一看有一种看不懂的感觉。。。
主要思路:
这道题可以用最小生成树来做,至于为什么,首先因为数据范围并不大,且要联通,那么我们可以把每一个矩形区域都标一个号,最开始是单个点,把边数存起来,然后用Kruskal求解。
由于数据还是有点大,如果直接暴力求点会导致sort超时,但我们可以观察题目中给予的图,同种篱笆每一水平方向的长度都相同,竖直方向也是这样,那么我们就可以考虑只存一条边,这样就可以优化了!
可是,如何建点呢,最开始一直没想出来,知道听了DY讲了后才懂,画个图就好看了
画的有点丑,请见谅……
这是题目的样例的图
中间的数字我们把它当成一个节点,这里面的边就是图中的边,那么我们就只存(n+1) * (m+1)条边。
我们还需要一种优化:
当我们找到某条边最短时,由于这条边代表的是某行或某列,我们就可以去搜对应的整行或整列,若两个点未联通,那么就可以把这条篱笆给拆掉,因为排序后这肯定是第一小的边使这两个点联通,故就用它。
代码:
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
const int MAXN = 2003;
int n , m , A , B;
long long ans;
int a[MAXN] , b[MAXN];
int x[MAXN] , y[MAXN];
int fa[MAXN * MAXN ];
void read( int &x ){
int f = 1; x = 0;
char s = getchar();
while( s < '0' || s > '9' ){
if( s == '-' )
f = -1;
s = getchar();
}
while( s >= '0' && s <= '9' ){
x = x * 10 + s - '0';
s = getchar();
}
x *= f;
}
int findSet( int x ){
if( x != fa[x] )
fa[x] = findSet( fa[x] );
return fa[x];
}
void makeSet( int x ){
for( int i = 1 ; i <= x; i ++ )
fa[i] = i;
}
void unionSet( int x , int y , int sum ){
int u = findSet( x ) , v = findSet( y );
if( u == v )
return ;
else{
ans += sum;
fa[u] = v;
}
}
int main()
{
read( A );read( B );read( n );read( m );
for( int i = 1 ; i <= n ; i++ )
read( a[i] );
for( int j = 1 ; j <= m ; j += 1 )
read( b[j] );
sort( a + 1 , a + n + 1 );//注意,这里输入顺序并没有升序输入,先排序
sort( b + 1 , b + m + 1 );
n ++;
m ++;
a[n] = A;
b[m] = B;//由于我们要求最后的一条边,那么就用最底边去剪
for( int i = 1; i <= n ; i += 1 )
x[i] = a[i] - a[i - 1];//现在来算边
for( int i = 1; i <= m ; i ++ )
y[i] = b[i] - b[i - 1];
sort( x + 1 , x + n + 1 );
sort( y + 1 , y + m + 1 );//后面这是Kruskal加优化
int i = 1 , j = 1;
y[ m + 1 ] = 0x3f3f3f3f;
makeSet( n * m );
while( i <= n || j <= m ){
if( x[i] <= y[j] && i <= n ){//若是它代表的是一行。
int k = ( i - 1 ) * m + 1;//从这一行的第一个点开始判断
for( int l = 1 ; l < m ; l ++ , k ++ )
unionSet( k , k + 1 , x[i] );//合并
i ++;
}
else{
int k = j;//若是每列,因为我们标号的顺序,所以它就是j
for( int l = 1 ; l < n ; l ++ , k += m )
unionSet( k , k + m , y[j] );
j ++;
}
}
printf( "%lld" , ans );//最后输出
return 0;
}