HYSBZ 1196 BZOJ 1196 最小生成树 贪心

题目描述：

OI island是一个非常漂亮的岛屿,自开发以来,到这儿来旅游的人很多。然而，由于该岛屿刚刚开发不久，所以那里的交通情况还是很糟糕。所以，OIER Association组织成立了，旨在建立OI island的交通系统。 OI island有n个旅游景点，不妨将它们从1到n标号。现在，OIER Association需要修公路将这些景点连接起来。一条公路连接两个景点。公路有，不妨称它们为一级公路和二级公路。一级公路上的车速快，但是修路的花费要大一些。 OIER Association打算修n-1条公路将这些景点连接起来（使得任意两个景点之间都会有一条路径）。为了保证公路系统的效率, OIER Association希望在这n-1条公路之中,至少有k条(0≤k≤n-1)一级公路。OIER Association也不希望为一条公路花费的钱。所以，他们希望在满足上述条件的情况下，花费最多的一条公路的花费尽可能的少。而你的任务就是，在给定一些可能修建的公路的情况下，选择n-1条公路，满足上面的条件.

Input

第一行有三个数n(1≤n≤10000),k(0≤k≤n-1),m(n-1≤m≤20000)，这些数之间用空格分开。 N和k如前所述，m表示有m对景点之间可以修公路。以下的m-1行，每一行有4个正整数a,b,c1,c2 （1≤a,b≤n,a≠b,1≤c2≤c1≤30000）表示在景点a与b 之间可以修公路,如果修一级公路,则需要c1的花费,如果修二级公路,则需要c2的花费。

Output

一个数据，表示花费最大的公路的花费。

Sample Input

10 4 203 9 6 31 3 4 15 3 10 28 9 8 76 8 8 37 1 3 24 9 9 510 8 9 12 6 9 16 7 9 82 6 2 13 8 9 53 2 9 61 6 10 35 6 3 12 7 6 17 8 6 210 9 2 17 1 10 2Sample Output

5

Hint

这道题目是求最小生成树中权值最大的边，但是其中必须要有 K 条一级路，这样就需要先按照 c1 排序，先构造 K 条最小生成树的边，然后取出没被使用过的边，重新按照 c2 从小到大排序。这样做的原因是先选的 K 条边是满足条件的按照 c1 最小的，之后再选的边也都是满足条件的按照 c2最小的，两次选边都把最小的囊括进来了，而且恰好可以生成最小生成树，所以只要求最小生成树中权值最大的边，这条边也是生成树中最小的边。也可以逆推，最小生成树一定存在，而且一部分来自 k , 另一部分来自后来选的边，这两部分是一定可以构成最小生成树的，由于这两部分都是最小，最小生成树中最大的边也是最小的。但是，后来我发现不用这么麻烦。

先给出原来的.

#include <algorithm>  
#include <iostream>  
#include <cstring>  
#include <cstdio>  
using namespace std ;  
int n , k , m , i , one , two ;  
int father[10005] ;  
struct Node{  
    int src , des , c1 , c2 ;
} Edge[20005] ;
bool cmp1( const Node &a , const Node &b ) { return a.c1 < b.c1 ; }  
bool cmp2( const Node &a , const Node &b ) { return a.c2 < b.c2 ; }   
  
int find( int u ){  
    return u == father[u] ? u : father[u] = find( father[u] ) ;  // 并查集找祖先  
}  

int main(){  
    scanf( "%d%d%d" , &n , &k , &m ) ;  
    for( i = 0 ; i <= n ; ++i )   
        father[i] = i ;
    for( i = 1 ; i <= m-1 ; ++i )  
        scanf( "%d%d%d%d" , &Edge[i].src , &Edge[i].des , &Edge[i].c1 , &Edge[i].c2 ) ;  
    sort( Edge+1 , Edge+m , cmp1 ) ;             // 先按照 c1 从小到大排序，选出必须的 k 条边  
    int ans = 0 , cnt = 0 ;  
    for( i = 1 ; i <= m-1 ; ++i )
        if( find( Edge[i].src ) != find( Edge[i].des ) ){
 			father[father[Edge[i].des]] = father[Edge[i].src] ;  
            ans = max( ans , Edge[i].c1 ) ;  
            if( ++cnt == k ) break ;    // 选出 k 条边了，再选剩下的  
        }  
    if( k == n-1 ){                     // 如果这时候已经是最小生成树，直接打印  
        cout << ans << endl ;  
        return 0 ;  
    }  
    sort( Edge+1 , Edge+m , cmp2 ) ;  
    for( i = 1 ; i <= m-1 ; ++i )  
        if( find( Edge[i].src ) != find( Edge[i].des ) ){
 			father[father[Edge[i].des]] = father[Edge[i].src] ;  // 继续选剩下的，构成最小生成树  
            ans = max( ans , Edge[i].c2 ) ;  
            if( ++cnt == n-1 ) break ;  
        }  
    cout << ans << endl ;  
    return 0 ;  
}  

上面的我考虑了，要取出之前求 K 条边剩余的边，其实这里边有多余，因为在我选出前 K 条之前没被选上的边，都是不合法的，会形成环，所以前面没被选上的不用考虑，直接考虑选了 K 条边之后的边。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std ;
int n , k , m , i , one , two ;
int father[10005] ;
struct Node{
	int src , des , c1 , c2 ;
} Edge[20005] ;
bool cmp1( const Node &a , const Node &b ) { return a.c1 < b.c1 ; }
bool cmp2( const Node &a , const Node &b ) { return a.c2 < b.c2 ; } 

int find( int u ){
	return u == father[u] ? u : father[u] = find( father[u] ) ;
}

int Merge( int a , int b ){
	one = find( a ) ;
	two = find( b ) ;
	if( one != two ){
		father[two] = one ;
		return 1 ;
	}
	return 0 ;
}

int main(){
	scanf( "%d%d%d" , &n , &k , &m ) ;
	for( i = 0 ; i <= n ; ++i ) 
		father[i] = i ;
	for( i = 1 ; i <= m-1 ; ++i )
		scanf( "%d%d%d%d" , &Edge[i].src , &Edge[i].des , &Edge[i].c1 , &Edge[i].c2 ) ;
	sort( Edge+1 , Edge+m , cmp1 ) ;
	int ans = 0 , temp , cnt = 0 ;
	for( i = 1 ; i <= m-1 ; ++i )
		if( Merge( Edge[i].src , Edge[i].des ) ){
			temp = ++cnt <= k ? Edge[i].c1 : Edge[i].c2 ,  
		    ans = max( ans , temp ) ;
			if( cnt == n-1 ) break ;
			if( cnt == k )
				sort( Edge+i+2 , Edge+m , cmp2 ) ;
		}
	cout << ans << endl ;
	return 0 ;
}

做完这题后，我想用 Prim 再做尝试，发现挺麻烦的，堆优化的比较需要改变，如果自己写一个优先级队列，耗点时间。之后我上网搜这道题目，想学习大佬们的解法，发现搜出一大堆二分+并查集......我的确没想到二分，不过我感觉我的做法也没错，我的只需要一次 Kruskal , 二分需要多次 Kruskal 求最小生成树

找出边权值的上下界，然后二分搜索最小的满足条件的解。

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
using namespace std ;
int n , k , m , i , one , two , ans ;
int father[10005] ;
struct Node{
	int src , des , c ;
	void Init( const Node &b ){
		src = b.src , des = b.des ;
	}
} Edge[20005] , Other[20005] ;    // c1 , c2 分开来存储，这样只需要一次排序，不用每次都排序
bool cmp( const Node &a , const Node &b ) { return a.c < b.c ; }

int find( int u ){
	return u == father[u] ? u : father[u] = find( father[u] ) ;
}

int Solve( Node *a , int &cnt , int mid , int k ){    // 是否可以构造 k 条边的最小生成树
	int u , v , w ;
	for( i = 1 ; i <= m-1 ; ++i ){
		u = a[i].src , v = a[i].des , w = a[i].c ;
		if( w > mid )                // 已经排好序了，w > mid , 意味着 mid 不足以容纳
			return 0 ;
		if( find( u ) != find( v ) ){
			father[father[v]] = father[u] ;
			if( ++cnt == k ) break ;
		}
	}
	return cnt == k ;  // 如果可以构造
}

int main(){
	scanf( "%d%d%d" , &n , &k , &m ) ;
	int l = 30005 , r = -1 , mid ;
	for( i = 1 ; i <= m-1 ; ++i ){
		scanf( "%d%d%d%d" , &Edge[i].src , &Edge[i].des , &Edge[i].c , &Other[i].c ) ;
		Other[i].Init( Edge[i] ) ;
		l = min( l , Other[i].c ) ;   // 找出下界
		r = max( r , Edge[i].c ) ;    // 找出上界
	}
	sort( Edge+1 , Edge+m , cmp ) ;
	sort( Other+1 , Other+m , cmp ) ;    // 分开来排序，就只需要一次排序
	while( l <= r ){
		mid = ( l + r ) / 2 ;
		for( i = 1 ; i <= n ; ++i ) 
		    father[i] = i ;
	    int cnt = 0 ;
		if( Solve( Edge , cnt , mid , k ) && Solve( Other , cnt , mid , n-1 ) ) 
			r = mid - 1 , ans = mid ;
		else
			l = mid + 1 ;
	}
	cout << ans << endl ;
	return 0 ;
}