BZOJ 1601: [Usaco2008 Oct]灌水( MST )

MST , kruskal 直接跑 

----------------------------------------------------------------------

#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstring>
#include<iostream>
 
#define rep( i , n ) for( int i = 0 ; i < n ; ++i ) 
#define clr( x , c ) memset( x , c , sizeof( x ) )
#define Rep( i , n ) for( int i = 1 ; i <= n ; ++i )
 
using namespace std;
 
const int maxn = 300 + 5;
 
struct edge {
int u , v , w;
bool operator < ( const edge &e ) const {
return w < e.w;
}
};
 
vector< edge > E;
 
int p[ maxn ];
 
int find( int x ) {
return x == p[ x ] ? x : p[ x ] = find( p[ x ] );
}
 
long long kruskal( int n ) {
rep( i , n ) p[ i ] = i;
sort( E.begin() , E.end() );
long long ans = 0;
rep( i , E.size() ) {
edge* e = &E[ i ];
int a = find( e -> u ) , b = find( e -> v );
if( a != b ) 
   p[ a ] = b , ans += e -> w;
   
}
return ans;
}
 
int main() {
// freopen( "test.in" , "r" , stdin );
int n;
cin >> n;
Rep( i , n ) {
int w;
scanf( "%d" , &w );
E.push_back( ( edge ) { 0 , i , w } );
}
Rep( i , n )
   Rep( j , n ) {
   
    int w;
   
    scanf( "%d" , &w );
   
    if( i != j ) E.push_back( ( edge ) { i , j , w } );
   
   }
cout << kruskal( n + 1 ) << "\n";
return 0;
}

 

---------------------------------------------------------------------- 

posted @ 2015-06-03 21:36  JSZX11556  阅读(117)  评论(0编辑  收藏  举报