一道比较妙的题,虽然正解思路和代码实现都很简单,但这个idea的质量确实是高。也验证了质量与难度不直接挂钩。比我出的某些屑题不知道高到哪里去了
与七桥问题有异曲同工之妙。打井实际上是将地下水与牧场相连,那么我们可以把地下水看做一个牧场,直接与地上的
n
n
n 个牧场连边。
于是我们惊讶地发现,问题变成了一个最小生成树的板子。Kruskal直接上就行。
真的很简单,不过我在点开题解之前一直想不明白,可能我还是需要大量地做思维题吧。
注意整张图是有 n + 1 n+1 n+1 个点的,所以最小生成树要连 n n n 条边而不是 n − 1 n-1 n−1 条
#include<cstdio>
#include<algorithm>
int n,cnt,x,y,ans,f[310];
struct node { int x,y,v; }E[91000];
inline int cmp(node p,node q) { return (p.v<q.v); }
int find(int x) { return f[x]==x?x:f[x]=find(f[x]); }
int main() {
scanf("%d",&n),cnt=n;
for (int i=1; i<=n; ++i)
scanf("%d",&x),E[i]={0,i,x},f[i]=i;
for (int i=1; i<=n; ++i) for (int j=1; j<=n; ++j)
scanf("%d",&x),i<j&&(E[++cnt]={i,j,x},0);
std::sort(E+1,E+cnt+1,cmp);
for (int i=1,t=0; t<n; ++i)
if ((x=find(E[i].x))!=(y=find(E[i].y)))
f[y]=x,ans+=E[i].v,++t;
printf("%d\n",ans);
return 0;
}