Description
发展采矿业当然首先得有矿井,小FF花了上次探险获得的千分之一的财富请人在岛上挖了n口矿井,但他似乎忘记考虑的矿井供电问题…
为了保证电力的供应,小FF 想到了两种办法:
1、在这一口矿井上建立一个发电站,费用为v(发电站的输出功率可以供给任意多个矿井)。
2、将这口矿井不另外的已经有电力供应的矿井之间建立电网,费用为p。
小FF希望身为“NewBe_One”计划首席工程师的你帮他想出一个保证所有矿井电力供应的最小花费。
Input
第一行一个整数n,表示矿井总数。
第2~n+1行,每行一个整数,第i个数v[i]表示在第i口矿井上建立发电站的费用。
接下来为一个n*n的矩阵P,其中p[i,j]表示在第i口矿井和第j口矿井之间建立电网的费用(数据保证有p[i,j]=p[j,i],且p[i,i]=0)。
Output
仅一个整数, 表示让所有矿井获得充足电能的最小花费。
Sample Input
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
Sample Output
9
Hint
【输出样例说明】小FF可以选择在4号矿井建立发电站然后把所有矿井都不其建立电网,总花费是3+2+2+2=9。
【数据范围】
对于30%的数据:1<=n<=50;
对于100%的数据:1<=n<=300;0<=v[i], p[i,j] <=10^5。
最小生成树啦
虚拟一个零点向每个矿井连边,权值为建立发电站的费用
然后跑prim或者kruskal
#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#define INF 100000000;
using namespace std;
int n,ans,d[1001],g[1001][1001],s[1001];
void prim(int v0)
{
int i,j,k,minn;
memset(s,0,sizeof(s));
for(i=0;i<=n;i++) d[i]=INF;
d[v0]=0;
ans=0;
for(i=0;i<=n;i++)
{
minn=INF;
k=1;
for(j=0;j<=n;j++)
if(!s[j]&&minn>d[j]) minn=d[j],k=j;
s[k]=1;
ans+=d[k];
for(j=0;j<=n;j++)
if(!s[j]&&d[j]>g[k][j]) d[j]=g[k][j];
}
}
int main()
{
int i,j,v;
cin>>n;
for(i=1;i<=n;i++) cin>>v,g[0][i]=v,g[i][0]=v;
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
cin>>g[i][j],g[j][i]=g[i][j];
prim(0);
cout<<ans;
return 0;
}
/*
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
*/