P1550 [USACO08OCT]Watering Hole G

每个点向井连边,边权为其点权。跑MST就行力。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
struct edge{
	int u,v,w;
	bool operator <(const edge &b)const{return w<b.w;}
}e[maxn];
int cnt=0;
int n;
int fa[305];
int g[305][305];
int Find(int x){
	return x==fa[x]?x:fa[x]=Find(fa[x]);
}
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		int x;cin>>x;
		e[++cnt].u=i,e[cnt].v=n+1,e[cnt].w=x;
	}
	for(int i=1;i<=n;i++)for(int j=1;j<=n;j++)cin>>g[i][j];
	for(int i=1;i<n;i++){
		for(int j=i+1;j<=n;j++){
			e[++cnt].u=i,e[cnt].v=j,e[cnt].w=g[i][j];
		}
	}
	for(int i=1;i<=n+1;i++)fa[i]=i;
	sort(e+1,e+cnt+1);
	int W=0;
	for(int i=1;i<=cnt;i++){
		if(Find(e[i].u)==Find(e[i].v))continue;
		fa[Find(e[i].u)]=Find(e[i].v);
		W+=e[i].w;
	}
	cout<<W;
	return 0;
}
上一篇:数据结构与算法--二叉堆(最大堆,最小堆)实现及原理


下一篇:f2f2 学习笔记 之一