[50Tproject-13]

来源:互联网 发布:知之阅读 pdf 编辑:程序博客网 时间:2024/05/22 09:50

题解:每条边的权值是边权*2+两个点的权值,跑一下最小生成树,因为要选择一个牧场过夜,所以tot还要加上最小的点的权值

#include <iostream>#include <cstdio>#include <algorithm>#include <climits> using namespace std;#define INF INT_MAXconst int M=10005;int n,m,x,y,z,tot=INF;int w[M],f[M],r[M*10];int u[M*10],v[M*10],val[M*10];void init(){    cin>>n>>m;    for(int i=1;i<=n;i++)    scanf("%d",&w[i]),tot=min(tot,w[i]);    for(int i=1;i<=m;i++)    {        scanf("%d%d%d",&u[i],&v[i],&z);        val[i]=(z<<1)+w[u[i]]+w[v[i]];    }}int find(int x){    return f[x]==x?x:f[x]=find(f[x]);}bool cmp(int a,int b){    return val[a]<val[b];}void kruskal(){    for(int i=1;i<=n;i++) f[i]=i;    for(int i=1;i<=m;i++) r[i]=i;    sort(r+1,r+1+m,cmp);    for(int i=1;i<=m;i++)    {        int e=r[i];        x=find(u[e]),y=find(v[e]);        if(x!=y){            tot+=val[e];            f[x]=y;        }    }}void work(){    kruskal();    printf("%d\n",tot);} int main(){    init();    work();    return 0;}
0 0