hdu 2489 Minimal Ratio Tree 最小生成树kruskal

这道题在大一团队训练赛上见过,很多人一次过了,是一道dfs+最小生成树,王昊天学长讲的时候也是三言两语带过,我有点不确定哈理工的水平什么·时候这么高了。。

组合dfs取出m个点,用这m个点构建最小生成树,这m个点的权值和是一定的,故边权值最小,则比才会最小;

思路:在Edge中选最小的,若这两点有一个不是m个点中的,就不把它放入树中;以为自己过不了,结果一次ac;

这道题有很多人卡在权值比上,精度没把握好的就会wr,其实用乘法更好,解决了精度问题;

幸好我提前看了discuss,发现了坑和解决办法;

代码:

#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
int n,m;
const int maxn=400;
int edge_wei[maxn][maxn];
int node_wei[maxn];
int fa[maxn];
int visit[maxn];
int k[maxn];
int nw;
int ew;
int start;
int shu;
struct Edge
{
    int u;
    int v;
    int weight;
};
Edge a[maxn];
void init()
{
    int i;
    for(i=1;i<=n;i++)
        fa[i]=i;
}
int find(int u)
{
    return fa[u]==u?fa[u]:fa[u]=find(fa[u]);
}
void join(int u,int v)
{
    fa[find(u)]=find(v);
}
bool cmp(Edge x,Edge y)
{
    return x.weight<y.weight;
}
void kruskal()
{

    init();
    int i,j;
    int cnt=0;
    int nw1=0;
    int ew1=0;
    /*for(i=1;i<=n;i++)
    {
        if(visit[i]==1)
            cout<<"i== "<<i<<endl;
    }*/
    for(i=1;i<=n;i++)
    {
        if(visit[i]==1)
            nw1+=node_wei[i];
    }
    for(i=0;i<start;i++)
    {
        int u=a[i].u;
        int v=a[i].v;
        if(visit[u]==visit[v]&&visit[u]==1&&find(u)!=find(v))
        {
            cnt++;
            ew1+=a[i].weight;
            join(u,v);
        }
        if(cnt==m-1)
            break;
    }
    if(nw*ew1<ew*nw1)
    {
        nw=nw1;
        ew=ew1;
         shu=0;
        for(i=1;i<=n;i++)
          {
             if(visit[i]==1)
             {
                k[shu++]=i;
             }
          }
    }
}
void select(int l,int p)
{
    int i;
    int sum=0;
    if(l==m+1)
    {
        kruskal();
    }
    for(i=p;i<=n;i++)
    {
        visit[i]=1;
        select(l+1,i+1);
        visit[i]=0;
    }
}
int main()
{
    while(cin>>n>>m)
    {
        if(n==0&&m==0)
            break;
        int i,j;
        memset(visit,0,sizeof(visit));
         start=0;
        for(i=1;i<=n;i++)
        {
            cin>>node_wei[i];
        }
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=n;j++)
            {
                cin>>edge_wei[i][j];
                if(j>=i)
                    continue;
                  a[start].u=i;
                  a[start].v=j;
                  a[start++].weight=edge_wei[i][j];
            }
        }
        sort(a,a+start,cmp);
         nw=1;
         ew=1500;
        select(1,1);
        for(i=0;i<m;i++)
        {
            if(i==0)
                cout<<k[i];
            else
                cout<<' '<<k[i];
        }
        cout<<endl;
    }
    return 0;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值