http://poj.org/problem?id=1639
题意:
给你一个N个点的图,限制其中一个点的度数必须要小于等于K,求满足条件的最小生成树。
思路:
对于那个限制点因为比较特殊,我们可以先求出不包括限制点的最小生成森林,这样除了限制点之外,整个图就会形成一个生成森林,假设森林的个数是v0个,那么由生成树的性质可以得出,限制点的度数应该至少为v0,接下去每次添加一条边(增加限制点的度数),然后删除一条边(这条边的选取可以用一个dfs实现)。这样就可以解决本题了。
代码:
#include <stdio.h>
#include <string.h>
const int VV = 21 ;
const int inf = (1 << 30) ;
int N ,K, cnt , park, NN;
int maze[VV][VV] ;
int dis[VV] ;
bool vis[VV] , mark[VV] , used[VV] ;
int mst , ans ;
int pre[VV] , best[VV] ,tree[VV][VV] , fa[VV];
char name[VV][11] ;
void dfs(int u , int ff ){
for(int v=1;v<=NN;v++){
if( tree[u][v]<inf && v!=ff && v!=u){
pre[v] = u ;
dfs( v , u ) ;
}
}
}
void prim( int s ){
int _min , n_m ;
vis[s] = 1 ;
memset( mark , 0 , sizeof(mark) );
for(int i=1;i<=NN;i++) dis[i] = inf ;
dis[s] = 0 ; fa[s] = s ;
for(int i=1;i<=NN;i++){
_min = inf ; n_m = -1 ;
for(int v = 1; v<=NN ; v++ ){
if( _min > dis[v] && !mark[v] ){
_min = dis[v] ; n_m = v ;
}
}
if( n_m == -1 ) break ;
vis[n_m] = mark[n_m] = 1 ;
tree[n_m][ fa[n_m] ] = tree[ fa[n_m] ][n_m] = _min ;
ans += _min ;
for(int v=1;v<=NN;v++){
if( maze[n_m][v]<inf && !mark[v] && dis[v]>maze[n_m][v])
dis[v] = maze[n_m][v] , fa[v] = n_m ;
}
}
_min = inf ; n_m = -1 ;
for(int i=1;i<=NN;i++){
if( mark[i] && maze[0][i] < _min )
_min = maze[0][i] , n_m = i ;
}
tree[0][n_m] = tree[n_m][0] = _min ;
ans += _min ;
used[ n_m ] = 1 ;
pre[ n_m ] = 0 ;
dfs( n_m , 0 ) ;
}
bool Best(int u, int ff ){
if( pre[u] == 0 ){
best[u] = 0 ; return true ;
}
if( best[u] != -1 ) return true ;
for(int v = 0 ; v <= NN ; v ++ ){
if( tree[u][v] < inf && v!=u && v!=ff ){
if( Best( v , u ) ){
if( best[v] != 0 ){
if( tree[ best[v] ][ pre[ best[v] ] ] > tree[u][v] ){
best[u] = best[v] ;
}
else
best[u] = u ;
}
else{
best[u] = u ;
}
pre[u] = v ;
return true ;
}
}
}
return false ;
}
int solve(){
memset( vis, 0 , sizeof(vis) ) ; mst = 0 ; ans = 0;
memset( used , 0 ,sizeof(used) ) ;
for(int i=0;i<=NN;i++) for(int j=0;j<=NN;j++) tree[i][j] = inf ;
for(int i=1;i<=NN;i++){
if( !vis[i] ){
prim(i) ;
mst ++ ;
}
}
if( mst > K ) return -1 ;
int res = ans , front = ans ;
for(int i=mst+1;i<=K && i<=NN; i++ ){
memset( best , -1, sizeof(best) );
for(int j=1;j<=NN;j++){
if( pre[j]!=0 && best[j] == -1 )
Best( j , -1 ) ;
}
int _min = inf , n_m ;
for(int j=1;j<=NN;j++){
if( maze[0][j]<inf && !used[j] ){
int rr = maze[0][j] - maze[ best[j] ][ pre[ best[j] ] ] ;
if( rr < _min ) _min = rr ,n_m = j ;
}
}
if( res > front + _min ) {
res = front + _min ;
}
front += _min ;
int u = best[ n_m ], v = pre[u] ;
tree[v][u] = tree[u][v] = inf ;
tree[n_m][0] = tree[0][n_m] = maze[0][n_m] ;
used[ n_m ] = 1 ; pre[n_m] = 0 ;
}
return res ;
}
int get_id( char *ch ){
for(int i=1;i<cnt;i++)
if( strcmp( name[i] , ch) == 0 ) return i ;
strcpy( name[cnt++] , ch );
return cnt - 1 ;
}
int main(){
char a[11] , b[11] ;
int c ;
strcpy( name[0] , "Park" ) ;
while( scanf("%d",&N) == 1 ){
for(int i=0;i<VV;i++) for(int j=0;j<VV;j++) maze[i][j] = inf ;
cnt = 1 ;
for(int i=0;i<N;i++){
scanf("%s%s%d",a,b,&c) ;
int na , nb ;
if( strcmp(a , "Park") == 0 ) na = 0 ;
else na = get_id( a ) ;
if( strcmp(b , "Park") == 0 ) nb = 0 ;
else nb = get_id( b ) ;
if( maze[na][nb] > c ) maze[na][nb] = maze[nb][na] = c ;
}
NN = cnt - 1;
scanf("%d",&K);
int res = solve();
if( res == -1 ) puts("ERROR");
else printf("Total miles driven: %d\n",res) ;
}
return 0 ;
}