2008年北京现场赛真题
没什么好说的,套两个板,一个枚举组合的板,一个最小生成树的板。就OK了
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<iomanip>
#define INF 99999999
using namespace std;
const int MAX=15+5;
int edge[MAX][MAX],dist[MAX],node[MAX];//node记录最终选的点
int vale[MAX],temp[MAX],n,m;//temp记录选的m个点
double minratio;
bool mark[MAX];
int Prim(int s){
int sum=0;
for(int i=1;i<=m;++i)mark[temp[i]]=false,dist[temp[i]]=edge[s][temp[i]];
mark[s]=true;
dist[s]=0;
for(int i=1;i<m;++i){
int point=s;
for(int j=1;j<=m;++j){
if(point == s && !mark[temp[j]])point=temp[j];
if(!mark[temp[j]] && dist[point]>dist[temp[j]])point=temp[j];
}
mark[point]=true;
sum+=dist[point];
for(int j=1;j<=m;++j){
if(!mark[temp[j]] && edge[point][temp[j]]<dist[temp[j]])dist[temp[j]]=edge[point][temp[j]];
}
}
return sum;
}
void dfs(int k,int num){
if(num == m){
int ans=0;
for(int i=1;i<=m;++i)ans+=vale[temp[i]];
double sum=Prim(k)*1.0/ans;
if(sum-minratio<-(1e-9)){
minratio=sum;
for(int i=1;i<=m;++i)node[i]=temp[i];
}
return;
}
if(n-k+num<m)return;
for(int i=k+1;i<=n;++i){
temp[num+1]=i;
dfs(i,num+1);
}
}
int main(){
while(cin>>n>>m,n+m){
minratio=INF*1.0;
for(int i=1;i<=n;++i)cin>>vale[i];
for(int i=1;i<=n;++i){
for(int j=1;j<=n;++j){
cin>>edge[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;
}