HDU 2489 Minimal Ratio Tree(dfs+最小生成树-Prim)

125 篇文章 0 订阅
28 篇文章 0 订阅

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;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值