Description
定义一棵树的系数ratio=边权和/点权和,现在给出一个n个点的完全图,问其只有m个点的子树的系数最小值
Input
多组用例,每组用例第一行输入两个正数n和m,之后输入n个整数val[i]表示第i个点的点权,最后输入一个n*n矩阵A表示该图的邻接矩阵(2<=m<=n<=15,点权边权范围都在[1,100]之间)
Output
输出系数最小的子树的m个节点,如果有多重方案则输出字典序最小的
Sample Input
3 2
30 20 10
0 6 2
6 0 3
2 3 0
2 2
1 1
0 2
2 0
0 0
Sample Output
1 3
1 2
Solution
对于固定的m个点,点权和是确定的,要想系数最小需要边权和最小,即为求这m个点的MST,dfs搜索所有m个点的情况,对每种情况prim跑一遍求出MST更新答案即可
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 22
int n,m,sum1,sum2,cost[maxn][maxn],val[maxn],p[maxn],ans[maxn];
int mincost[maxn];//从集合X出发的边到每个顶点的最小权值
bool used[maxn];//顶点i是否包含在集合X中
int prim(int V)
{
for(int i=0;i<V;i++)//初始化
{
mincost[i]=cost[p[0]][p[i]];
used[i]=0;
}
used[0]=1;//把第一个顶点加进集合X中
int res=0;
for(int i=1;i<V;i++)
{
int v=-1,minc=INF;
for(int u=0;u<V;u++)//从不属于集合X的顶点中选取从X到其权值最小的顶点
if(!used[u]&&mincost[u]<minc)
v=u,minc=mincost[u];
if(v==-1)break;
used[v]=1;//把顶点v加到集合X中
res+=minc;//把边权加到结果中
for(int u=0;u<V;u++)//mincost中始终存的是集合X中元素到每个顶点的最小权值
if(!used[u])
mincost[u]=min(mincost[u],cost[p[u]][p[v]]);
}
return res;
}
void dfs(int pos,int num)
{
if(num==m)
{
int tsum1=prim(m),tsum2=0;
for(int i=0;i<m;i++)tsum2+=val[p[i]];
if(sum2==-1||tsum1*sum2<tsum2*sum1)
{
sum1=tsum1,sum2=tsum2;
for(int i=0;i<m;i++)ans[i]=p[i];
}
}
if(n-pos+1+num<m)return ;
for(int i=pos;i<=n;i++)
{
p[num]=i;
dfs(i+1,num+1);
}
}
int main()
{
while(~scanf("%d%d",&n,&m),n||m)
{
sum2=-1;
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",&cost[i][j]);
dfs(1,0);
for(int i=0;i<m;i++)printf("%d%c",ans[i],i==m-1?'\n':' ');
}
return 0;
}