[USACO08OCT]Watering Hole

[USACO08OCT]Watering Hole

题目大意：

Farmer John 有(n(nle300))个牧场，他希望灌溉他的所有牧场。牧场编号为(1sim n)，要灌溉一个牧场有两种方式，一个是直接在这个牧场建设一个小型水库，另一个是从别的牧场向这个牧场引水。在第(i)个牧场建立小型水库需要(W_i)美元，而从第(i)个牧场向第(j)个牧场引水需要(P_{i,j})美元。即便牧场(j)没有建设小型水库，只要有别的有水的水库向它引水，那它自己也有水，而且可以向别的水库引水。请告诉 FJ 灌溉所有牧场所需的最小代价。

思路：

(P_i)相当于连向虚点(0)的边，然后直接Kruskal即可。

源代码：

#include<cstdio>
#include<cctype>
#include<algorithm>
inline int getint() {
	register char ch;
	while(!isdigit(ch=getchar()));
	register int x=ch^'0';
	while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
	return x;
}
const int N=301,M=45150;
struct Edge {
	int u,v,w;
	bool operator < (const Edge &rhs) const {
		return w<rhs.w;
	}
};
Edge e[M];
int m;
class DisjointSet {
	private:
		int anc[N];
		int find(const int &x) {
			return x==anc[x]?x:anc[x]=find(anc[x]);
		}
	public:
		void reset(const int &n) {
			for(register int i=0;i<=n;i++) anc[i]=i;
		}
		void merge(const int &x,const int &y) {
			anc[find(x)]=find(y);
		}
		bool same(const int &x,const int &y) {
			return find(x)==find(y);
		}
};
DisjointSet s;
int main() {
	const int n=getint();
	for(register int i=1;i<=n;i++) {
		e[m++]=(Edge){0,i,getint()};
	}
	for(register int i=1;i<=n;i++) {
		for(register int j=1;j<=n;j++) {
			const int w=getint();
			if(j<i) e[m++]=(Edge){i,j,w};
		}
	}
	s.reset(n);
	std::sort(&e[0],&e[m]);
	int ans=0;
	for(register int i=0;i<m;i++) {
		const int &u=e[i].u,&v=e[i].v,&w=e[i].w;
		if(s.same(u,v)) continue;
		s.merge(u,v);
		ans+=w;
	}
	printf("%d
",ans);
	return 0;
}