首页 > 代码库 > ZOJ 1586 QS Network prim优化模板
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 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; }
ZOJ 1586 QS Network prim优化模板
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。