hdu 2489 Minimal Ratio Tree(prim+dfs)

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

算最小Ratio，edge weight可以由prim算出，同时记录temp[ i ]得到ans，然后dfs暴力枚举n个点中m个点的最小ratio0。

这里有一个枝剪，如n=5，m=3，（3,4,5）那么第一个次只需要遍历3个数就好了，第二次四个。。。所以要num<=n-m+k.

/*
prim应用错误：是将m个点生成最小树，而不是那n个！
temp[i]记录的是编号；

*/
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<iomanip>
using namespace std;

const int maxn=20;
const int maxcost=999999999;
const int inf=999999999;
double minratio;
int m,node[maxn],val[maxn],temp[maxn];
int graph[maxn][maxn],lowcost[maxn],n;

int prim(int s){
    int lowcost[maxn];
    int i,j,minn,minid,sum=0;
    for(int i=1;i<=m;i++)
        lowcost[temp[i]]=graph[s][temp[i]];
    lowcost[s]=-1;
    for(int i=2;i<=m;i++){
        int minn=inf;
        int minid=0;
        for(int j=1;j<=m;j++){
            if(lowcost[temp[j]]!=-1&&lowcost[temp[j]]<minn){
                minid=temp[j];
                minn=lowcost[temp[j]];
            }
        }
        sum+=minn;
        lowcost[minid]=-1;
        for(int j=1;j<=m;j++)
            if(lowcost[temp[j]]>graph[minid][temp[j]])
                lowcost[temp[j]]=graph[minid][temp[j]];
    }
    return sum;
}

void dfs(int k,int num){
    if(num==m){
        int ans=0;
        for(int i=1;i<=m;i++)
            ans+=val[temp[i]];
        double sum=prim(k)*1.0/ans;
        if(sum<minratio){
            minratio=sum;
            for(int i=1;i<=m;++i)node[i]=temp[i];
        }
        return;
    }
    if(n-m+k<num)
        return;
    for(int i=k+1;i<=n;i++){
        temp[num+1]=i;
        dfs(i,num+1);
    }

}
int main(){
    while(scanf("%d%d",&n,&m)==2&&n+m){
        fill(graph[0],graph[0]+maxn*maxn,inf);
        minratio=inf*1.0;
        for(int i=1;i<=n;i++)
            scanf("%d",&val[i]);
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                scanf("%d",&graph[i][j]);
        for(int i=1;i<=n;i++){
            temp[1]=i;
            dfs(i,1);
        }
        for(int i=1;i<m;++i)cout<<node[i]<<' ';
        cout<<node[m]<<endl;
    }
    return 0;
}