ZOJ 1586 QS Network prim优化模板

链接:

1586




题意:

有一个N X N的网络,每两个点都有边相连,边的权值用邻接矩阵输入,每个点也有一个权值,当它们之间的那条边被选取时,需要加上两个点的权值。求这个网络的最小生成树。





直接套用prim算法的模板 其中用到一个节约内存的优化 将lowdistance 和visit 两个数组

结合起来 如果访问过lowdistance改成-1即可

另外该题中边的权值应为边本身的权值加上两端点的权值。





#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define maxx   9999999
int map[1010][1010];
int price[1010];
int lowdis[1010];
int n,ans;
void prim()
{
    int i,j,minn,loc;
    for(i=1; i<=n; i++)
        lowdis[i]=map[1][i];
    lowdis[1]=-1;
    for(i=1; i<n; i++)
    {
        minn=maxx;
        for(j=1; j<=n; j++)
            if(lowdis[j]!=-1&&lowdis[j]<minn)
                minn=lowdis[j],loc=j;
        ans+=lowdis[loc];
        lowdis[loc]=-1;
        for(j=1; j<=n; j++)
            if(lowdis[j]>map[loc][j])
                lowdis[j]=map[loc][j];
    }
    return;
}
int main()
{
    int t,i,j;
    scanf("%d",&t);
    while(t--)
    {
        ans=0;
        scanf("%d",&n);
        for(i=1; i<=n; i++)
            scanf("%d",&price[i]);
        for(i=1; i<=n; i++)
            for(j=1; j<=n; j++)
            {
                scanf("%d",&map[i][j]);
                if(i==j)
                    map[i][j]=maxx;
                else
                    map[i][j]+=price[i]+price[j];
            }
        prim();
        cout<<ans<<endl;
    }
    return 0;
}





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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值