首页 > 代码库 > BZOJ 1601 Usaco2008 Oct 灌水 Prim

BZOJ 1601 Usaco2008 Oct 灌水 Prim

题目大意:给定n个点,每个点可以花w[i]的代价建水井,或者花p[i][j]的代价连接到一个已经供水的点,求最小花销

将每个点向超级源连一条边,边权为w[i]

求最小生成树即可

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 310
using namespace std;
int n,map[M][M];
int Prim(int n)
{
	static int f[M];
	static bool v[M];
	memset(f,0x3f,sizeof f);
	f[1]=0;
	int i,j,k,re=0;
	for(j=1;j<=n;j++)
	{
		k=0;
		for(i=1;i<=n;i++)
			if(!v[i]&&f[i]<f[k])
				k=i;
		v[k]=1;re+=f[k];
		for(i=1;i<=n;i++)
			f[i]=min(f[i],map[k][i]);
	}
	return re;
}
int main()
{
	int i,j,x;
	cin>>n;
	for(i=1;i<=n;i++)
	{
		scanf("%d",&x);
		map[i][n+1]=map[n+1][i]=x;
	}
	for(i=1;i<=n;i++)
		for(j=1;j<=n;j++)
			scanf("%d",&map[i][j]);
	cout<<Prim(n+1)<<endl;
	return 0;
}


BZOJ 1601 Usaco2008 Oct 灌水 Prim