bzoj 1232: [Usaco2008Nov]安慰奶牛cheer

→题目链接←

读题就可以看出肯定用最小生成树

只按照边权找到最小生成树后，计算最后的时间是

Σ边权*2+Σ点权*出度+我们要选的过夜的点的权值

然后发现，如果只按照边权找最小生成树是不正确的

因为我们正常跑kruskal的时候每次会将边权*2+两边点的点权加进答案

所以我们可以在排序前把每条边两边点的点权都加进这个边权之中，并且把原边权*2

这样再跑kruskal就每次把边权加上就好了，最后的过夜点一定是点权最小的那个

代码：

#include<iostream>
#include<cstdio>
#include<vector>
#include<algorithm>
#include<queue>

#define ll long long
#define inf 999999999

using namespace std;

struct node{
	int s,t,len;
	friend bool operator < (node a,node b){
		return a.len<b.len;
	}
}e[100010];

int n,m;
int ans=0;
int fa[10010];
int w[10010];
int Min=inf;

int getroot(int x){
	if(fa[x]==x)return x;
	return fa[x]=getroot(fa[x]);
}

int main(){
	scanf("%d%d",&n,&m);
	for(int i=1; i<=n; i++)fa[i]=i;
	for(int i=1; i<=n; i++)scanf("%d",&w[i]),Min=min(Min,w[i]);
	for(int i=0; i<m; i++)scanf("%d%d%d",&e[i].s,&e[i].t,&e[i].len),e[i].len=e[i].len*2+w[e[i].s]+w[e[i].t];
	sort(e,e+m);
	int sum=1;
	for(int i=0; i<m; i++){
		int x=getroot(e[i].s);
		int y=getroot(e[i].t);
		if(x==y)continue;
		ans+=e[i].len;
		sum++;
		if(sum==n)break;
		fa[x]=y;
	}
	printf("%d\n",ans+Min);
	
	return 0;
}