标签:
这题还是用最小生成树的模板就可以解决,不过在边的值上面要稍微处理一下。
#include <iostream> #include <cstdio> #include <cstring> #include <algorithm> using namespace std; const int N=1000; const int inf=1<<28; int cost[N][N],mincost[N],brand[N]; bool used[N]; int n; int prim() { for(int i=1;i<=n;i++) { mincost[i]=inf; used[i]=false; } mincost[1]=0; int res=0; while(true) { int _min=inf,u=-1; for(int i=1;i<=n;i++) if(!used[i]&&_min>mincost[i]) _min=mincost[i],u=i; if(u==-1) break; used[u]=true; res+=_min; for(int i=1;i<=n;i++) mincost[i]=min(mincost[i],cost[u][i]); } return res; } int main() { int t; scanf("%d",&t); while(t--) { scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%d",&brand[i]); for(int i=1;i<=n;i++) for(int j=1;j<=n;j++) { scanf("%d",&cost[i][j]); cost[i][j]+=brand[i]+brand[j]; } int ans=prim(); printf("%d\n",ans); } return 0; }
标签:
原文地址:http://blog.csdn.net/u013621213/article/details/43835357