题目: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;
}