hdu 2489 MST+dfs

http://acm.hdu.edu.cn/showproblem.php?pid=2489

题目大意是有n个顶点的无向完全图，要你选m个点，使得m个点构成的图的m-1条边的权重之和比上m个点的值的和最小，即Sum(edge weight)/Sum(point weight)最小；

由于题目的数据很小，因此我就直接暴力了。。。

就是从n个点中先选m个点，对应一个最小值，然后对这m个点进行搜索就可以了。。。

如果一开始就从第一个点就开始dfs，那么就可以保证最后的结果就是题目要求的。。。

#include<iostream>
const int N=20;
const int inf=10000000;
using namespace std;

int node[N];//点的权值
int recode[N];//保存点
int edge[N][N];
int lowcost[N];
int map[N][N];//用来保存边
int path[N]; //保存路径
int n,m;
double MIN;

double  prim(int v0){
    double sum=0;
    for(int i=1;i<=m;i++){
        lowcost[i]=map[v0][i];
    }
    lowcost[v0]=-1;
    for(int i=1;i<m;i++){
        int min=inf,v=-1;
        for(int j=1;j<=m;j++){
            if(lowcost[j]!=-1&&lowcost[j]<min){
                v=j,min=lowcost[j];
            }
        }
        if(v!=-1){
            sum+=lowcost[v];
            lowcost[v]=-1;
            for(int k=1;k<=m;k++){
                if(lowcost[k]!=-1&&map[v][k]<lowcost[k]){
                    lowcost[k]=map[v][k];
                }
            }
        }
    }
    return sum;
}
void dfs(int v0,int count){
    recode[count]=v0;
    if(count==m){
        for(int i=1;i<=m;i++){
            map[i][i]=0;
            for(int j=i+1;j<=m;j++){
                map[i][j]=map[j][i]=inf;
            }
        }
        for(int i=1;i<=m;i++){
            for(int j=1;j<=m;j++){
                map[i][j]=edge[recode[i]][recode[j]];
            }
        }
        double s1=prim(1),s2=0;
        for(int i=1;i<=m;i++){
            s2+=node[recode[i]];
        }
        double s=s1/s2;
        if(s<MIN){
            MIN=s;
            //保存路径
            for(int i=1;i<=m;i++)
                path[i]=recode[i];
        }
    }
    for(int i=v0+1;i<=n;i++)
        dfs(i,count+1);
}

int main(){
    while(scanf("%d%d",&n,&m)!=EOF){
        if(n==0&&m==0)break;
        for(int i=1;i<=n;i++){
            scanf("%d",&node[i]);
        }
        for(int i=1;i<=n;i++){
            for(int j=1;j<=n;j++){
                scanf("%d",&edge[i][j]);
            }
        }
        MIN=1e18;
        for(int i=1;i<=n;i++)
            dfs(i,1);
        printf("%d",path[1]);
        for(int i=2;i<=m;i++){
            printf(" %d",path[i]);
        }
        printf("\n");
    }
    return 0;
}