【Luogu P1550】 [USACO08OCT]Watering Hole G

题目大意:

建造一个水库需要花费 \(w_i\),连接两块土地需要花费 \(P_{i,j}\),计算所有土地都有水时所需的最少代价。

正文:

\(0\) 处建立一个超级原点就可直接最小生成树。

代码:

int FIND(int x){return (x==fa[x])?x:fa[x]=FIND(fa[x]);}

void add(int u, int v, int w)
{
	e[++tot] = (edge){u, v, head[u], w}, head[u] = tot;
}

int cmp (edge a, edge b)
{
	return a.w < b.w;
}

int main()
{
	scanf ("%d", &n);
	for (int i = 0; i <= n; ++i) fa[i] = i;
	for (int i = 1; i <= n; ++i)
	{
		int w;
		scanf ("%d", &w);
		add(0, i, w);
	}
	for (register int i = 1; i <= n; ++i)
	{
		for (register int j = 1; j <= n; ++j)
		{
			int w;
			scanf ("%d", &w);
			if (i == j)continue;
			add(i, j, w);
		}
	}
	sort(e + 1, e + 1 + tot, cmp);
	for (int i = 1; i <= tot ; i++)
	{
		int f = FIND(e[i].from), t = FIND(e[i].to); 
		if (f == t) continue;
		fa[f] = t;
		ans += e[i].w * 1ll;
	}
	printf ("%lld", ans);
	return 0;
}
posted @ 2020-08-11 20:00  Jayun  阅读(56)  评论(0编辑  收藏  举报