1488:新的开始
【题目描述】
发展采矿业当然首先得有矿井,小 FF 花了上次探险获得的千分之一的财富请人在岛上挖了 n 口矿井,但他似乎忘记考虑的矿井供电问题……
为了保证电力的供应,小 FF 想到了两种办法:
在这一口矿井上建立一个发电站,费用为 v(发电站的输出功率可以供给任意多个矿井)。
将这口矿井与另外的已经有电力供应的矿井之间建立电网,费用为 p。
小 FF 希望身为「NewBe_One」计划首席工程师的你帮他想出一个保证所有矿井电力供应的最小花费。
【输入】
第一行一个整数 n,表示矿井总数。
第 2∼n+1 行,每行一个整数,第 i 个数 vi 表示在第 i 口矿井上建立发电站的费用。
接下来为一个 n×n 的矩阵 p,其中 pi,j 表示在第 i 口矿井和第 j 口矿井之间建立电网的费用(数据保证有pi,j=pj,i ,且 pi,i=0。
【输出】
输出仅一个整数,表示让所有矿井获得充足电能的最小花费。
【输入样例】
4
5
4
4
3
0 2 2 2
2 0 3 3
2 3 0 4
2 3 4 0
【输出样例】
9
【提示】
样例解释
小 FF 可以选择在 4 号矿井建立发电站然后把所有矿井都不其建立电网,总花费是 3+2+2+2=9。
数据范围:
对于 30% 的数据:1≤n≤50;
对于 100% 的数据:1≤n≤300,0≤vi,pi,j≤105 。
解题思路:
这就是一道Primt的模板题,求最小生成树。当建立电网的钱比建发电站的钱更多的时候,我们就要考虑多建一个发电站,那么我们可以建立一个虚拟的矿井,将虚拟矿井到其他矿井全部连接,距离为建发电站的费用,然后生成的最小生成树就答案。
#include<iostream>
#include<algorithm>
using namespace std;
int w[1001][1001];
int dis[1001]={0};
int n;
int Primt(int x)
{
int cost=0;
for(int i=0;i<=n;i++)
dis[i]=w[x][i];
dis[x]=0;
for(int i=1;i<=n;i++)
{
int t=-1;
int Min=1e9;
for(int j=0;j<=n;j++)
{
if(dis[j]!=0&&Min>dis[j])
{
Min=dis[j];
t=j;
}
}
if(t==-1)
return -1;
dis[t]=0;
cost+=Min;
for(int j=0;j<=n;j++)
{
if(dis[j]!=0&&dis[j]>w[t][j])
{
dis[j]=w[t][j];
}
}
}
return cost;
}
int main()
{
cin>>n;
//将第0口矿井当成虚拟矿井
w[0][0]=1e9;
for(int i=1;i<=n;i++)
{
cin>>w[0][i];
w[i][0]=w[0][i]; //第0口矿井到其他矿井的距离为发电站的费用
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
cin>>w[i][j];
if(w[i][j]==0)
w[i][j]=1e9;
}
}
cout<<Primt(0);
}