题目链接
刚看到这道题的时候,我相信很多读者会把它与二分答案的最优比例生成树联系在一起,当然,我也是这么想的,然而,后来怎么推也都没法合理的将它的优先队列的方程推导出来,所以想必是错了的。
这道题还的确是用最优比例生成树,因为它的N、M偏小,最多也就是15,那么搜索就可行了,我们可以用dfs()去搜,每次将搜索的节点放进一个预存的数组中,然后当恰好搜到M个节点的时候,我们就对这M个节点去建立最优比例生成树,然后比较其与其他方案的大小就行了。
#include <iostream>
#include <cstdio>
#include <cmath>
#include <string>
#include <cstring>
#include <algorithm>
#include <limits>
#include <vector>
#include <stack>
#include <queue>
#include <set>
#include <map>
#define lowbit(x) ( x&(-x) )
#define pi 3.141592653589793
#define e 2.718281828459045
#define INF 0x3f3f3f3f
#define efs 1e-3
using namespace std;
typedef unsigned long long ull;
typedef long long ll;
const int maxN=16;
int N, M;
double a[maxN], edge[maxN][maxN], lowercost[maxN], minn;
int need[maxN], ques[maxN];
double prime(int pos)
{
double len=0;
bool vis[maxN]; memset(vis, false, sizeof(vis));
vis[pos]=true;
for(int i=1; i<=M; i++)
{
lowercost[need[i]]=edge[need[pos]][need[i]];
}
for(int line=1; line<M; line++)
{
double tmp=INF;
for(int i=1; i<=M; i++)
{
if(!vis[i] && tmp > lowercost[need[i]])
{
tmp = lowercost[need[i]];
pos = i;
}
}
vis[pos] = true;
len += tmp;
for(int i=1; i<=M; i++)
{
lowercost[need[i]]=min(lowercost[need[i]], edge[need[pos]][need[i]]);
}
}
return len;
}
void dfs(int st, int cnt)
{
if(cnt == M)
{
double need_node=0;
for(int i=1; i<=M; i++) need_node += a[need[i]];
double tmp = prime(1)/need_node;
if(minn - tmp > efs)
{
minn = tmp;
for(int i=1; i<=M; i++) ques[i] = need[i];
}
}
else
{
for(int i=st+1; i<=N-M+cnt+1; i++)
{
need[cnt+1]=i;
dfs(i, cnt+1);
}
}
}
int main()
{
while(scanf("%d%d", &N, &M) && (N || M))
{
for(int i=1; i<=N; i++) scanf("%lf", &a[i]);
for(int i=1; i<=N; i++) for(int j=1; j<=N; j++) scanf("%lf", &edge[i][j]);
minn=INF;
for(int i=1; i<=N-M+1; i++)
{
need[1]=i;
dfs(i, 1);
}
for(int i=1; i<=M; i++) printf("%d%c", ques[i], i==M?'\n':' ' );
}
return 0;
}