Picnic Planning 最小度限制生成树

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 ;
}